DataDrameから必要なデータを取り出すための条件設定の方法

Question

実現したいこと

下記に示すようなDataFrameがあります。この中で、Technologyのカラムで値が"AAA"を含む行を取り出したい時、すなわち、ID=10とID=13とを抽出したい。

環境

• Windows10(64bit)
• Python (v3.9.4)
• Pandas (v1.2.4)
• Jupyter Notebook

DataFrameの構造

ID	Technology	Country
10	AAA	Japan
11	BBB	USA
13	AAA-BBB	Germany

試したこと

df[df["Technology"]=="AAA"]
で実行すると、ID=10の行しか取り出せません。

df["AAA" in df["Technology"]]
で実行すると、
KeyError : False
となってしまいます。

どのような記述をすればよいのでしょうか？

Answer 1

以下です。

>>> print(df)
   ID Technology  Country
0  10        AAA    Japan
1  11        BBB      USA
2  13    AAA-BBB  Germany
>>> print(df[df['Technology'].str.contains('AAA')])
   ID Technology  Country
0  10        AAA    Japan
2  13    AAA-BBB  Germany

Comments

[BuhKeil]

ありがとうございました！

Answer 2

こういう情報って、「panadas DataFrame 部分一致」あたりで検索すると出てきますよ。

こんな感じでしょうか。

df[df['Technology'].str.contains('AAA')]

ただし、含まれているでひっぱってくるので予期しない結果になる可能性もあります。 データの仕様によりますね。