【python】DataFrameのデータ取り出し
pandasとnumpyのインポート。
また、pandasからSeriesとDataFrameもインポートする。
import pandas as pd import numpy as np from pandas import Series, DataFrame
テスト用のDataFrameを作成
dataframe1 = DataFrame(np.arange(15).reshape(5,3), index=['A','B','C','D','E'], columns=['col1','col2','col3'])
インデックス・行を指定して取り出し
#行を指定して取り出し dataframe1.ix[1] #インデックスを指定して取り出し dataframe1.ix['C']
実行結果
#行を指定して取り出し col1 3 col2 4 col3 5 Name: B, dtype: int32 #インデックスを指定して取り出し col1 6 col2 7 col3 8 Name: C, dtype: int32
条件を満たすデータのみ取り出し
特定の列が値を満たす場合のみ取り出す
dataframe1[dataframe1['col2']>8]
dataframe1のcol2の値が8より大きい場合のみを対象にする。
実行結果
col1 col2 col3 D 9 10 11 E 12 13 14