ここではPandasにおけるDataFrameの列ラベルを抽出する方法について解説しています。
基本的な型
DataFrame.query('条件式')「query」を用いることで列の値に対して条件式を設定することで容易に抽出することが可能です。
使用例
f = pd.DataFrame({
"num":[0,1,2,3,4,5],
"type":["m","f","m","f","f","m"],
"flag":["_","|","_","|","|","_"]
})
print(f.query('num == 2'))
# flag num type
#2 _ 2 m上記のプログラムは条件式でnumが2である行だけを抽出しています。
csvファイルなどの大量のデータを読み込んだ際、必要な情報だけを抽出したい場合などに使用すると便利です。