DataFrameからlist内の要素を含む行を抽出する方法（もしくはbooleanで）

このようなリストとデータフレームがあったときに、下記のように返す書き方を考えているのですが、わかりません。
できればfor文以外で処理できればと考えています。

[True, True, True, True, False]

python
1list = ["a", "b", "c", "d"]
2
3print(df) 
4#    name　value
5# 0    a     A
6# 1    b     B
7# 2    c     C
8# 3    d     D
9# 4    e     E
10# 5    f     F

最終的には、このようなデータフレームを作成したいのですが、もっと簡単な方法があれば、教えてください。
よろしくお願いします。

#    name　value
# 0    a     A
# 1    b     B
# 2    c     C
# 3    d     D

ppaul

2021/07/31 12:29

もっと簡単な方法を求めるなら、今やっている方法を質問に追加してください。

kiyushi

2021/07/31 12:48

書き方が悪くてすみません。 booleanでの配列[True, True, True, True, False]を作ってから、元のデータフレームを切り出して、 # name　value # 0 a A # 1 b B # 2 c C # 3 d D このようなデータフレームワーク作る方法を検討していて、できずにいます。で、そもそもその方法よりもっといい方法があるのであれば、別解的に教えていただけたら嬉しいな、という感じでした。

行動規範の内容に同意します

回答1件

ベストアンサー

listは組込みクラス名なので、変数名として使うのはやめましょう。

方法1 (Bool値)

python
1>>> print(df)
2  name value
30  a    A
41  b    B
52  c    C
63  d    D
74  e    E
85  f    F
9>>> condition = ["a", "b", "c", "d"]
10>>> print(df['name'].apply(lambda x: x in condition))
110     True
121     True
132     True
143     True
154    False
165    False
17Name: name, dtype: bool
18>>> print(df[df['name'].apply(lambda x: x in condition)])
19  name value
200  a    A
211  b    B
222  c    C
233  d    D

方法2 (merge)

python
1>>> df2 = pd.DataFrame({'name': condition})
2>>> df3 = pd.merge(df2, df)
3>>> print(df3)
4  name value
50  a    A
61  b    B
72  c    C
83  d    D
9