強弱と高低

Python3をメインに

pandasで条件にあった行を抽出してさらに列を指定する

やりたいこと

ある巨大行列のなかで、特定の列の値がXである行を抽出する。そしてさらに利用する列を指定する。

コード

import pandas as pd
df = pd.read_csv("test.csv", header=None)

df[df[2] == X].ix[:,5:90]

df[2] == Xが第2列の値がXである行を抽出してくれる。
更に加えて、.ix[;5:8]をすることで抽出した行の中から5列から7列目までの列を取り出してくれる。

とても便利ですねpandas