df.query()の使い方のメモ

恥ずかしいけど、覚えた事を復習。

String	Number
a	0
i	1
u	2
e	3
o	4

こんな表(tmp.csvとする)があった時に、以下のようになる。

import pandas as pd

df = pd.to_csv("./tmp.csv")
df.query("1 <= Number <= 3")

String	Number
i	1
u	2
e	3

String列の文字列で抽出したい時は、条件全体は「'」でくくり、文字列自体は「"」でくくってあげると以下のようになる。

df.query('String == "a" or String == "i" ')

String	Number
a	0
i	1

糞なのは、以下。

ho = "a"
ge = "i"
df.query("String == @ho or String == @ge ")

いやー、普段SQLを書いていないのもバレるし、そもそも・・・って、恥をかきました。

がんばる。