Python使ったことないから知らんけど
インデクス付けて選択して抽出したら?