pandasのデータフレームで、ある列に他の列によって指定された文字が含まれる行だけを抽出したい

以下のようなデータフレームがあります。

df
1    　A     B     C
20     X     X     Y
31   X,Y,Z   Y     Z
42    Y,Z    Z     X
53     Z     Z     Z
64    X,Z    X     X
7...  ...   ...   ...
895   X,Y    Y     X
996   X,Z    X     X
1097    Y     Z     Y
1198  X,Y,Z   Y     Y
1299  X,Y,Z   X     Z

A列の文字列がB列、C列の文字をどちらも含んでいる行だけ抽出したいです。

df2
1    　A     B     C
21   X,Y,Z   Y     Z
33     Z     Z     Z
44    X,Z    X     X
56     X     X     X
69   X,Y,Z   X     Y  
7...  ...   ...   ...
893    Y     Y     Y
995   X,Y    Y     X
1096   X,Z    X     X
1198  X,Y,Z   Y     Y
1299  X,Y,Z   X     Z

以下のことを試しましたが、すべて失敗しました。

python
1df2 = df[df['A'].contains(df['B']) & df['A'].contains(df['C'])]
2AttributeError: 'Series' object has no attribute 'contains'
3
4df2 = df[df['A'].str.contains(df['B']) & df['A'].str.contains(df['C'])]
5TypeError: 'Series' objects are mutable, thus they cannot be hashed
6
7df2 = df[df['A'].contains(df['B'].str) & df['A'].contains(df['C'].str)]
8AttributeError: 'Series' object has no attribute 'contains'
9
10df2 = df[df['A'].str.contains(df['B'].str) & df['A'].str.contains(df['C'].str)]
11TypeError: first argument must be string or compiled pattern

よろしくお願いします。

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

回答1件

ベストアンサー

applyを使いましょう。

python
1>>> import pandas as pd
2>>>
3>>> df = pd.DataFrame({'A':['X,Y,Z', 'Z', 'X,Z','X','X,Y,Z'], 'B':['Y','Z','X','X','X'], 'C':['Z','Z','X','X','Y']})
4>>>
5>>> def func(row):
6...     return (row['B'] in row['A']) and (row['C'] in row['A'])
7...
8>>> df[df.apply(func, axis=1)]
9       A  B  C
100  X,Y,Z  Y  Z
111      Z  Z  Z
122    X,Z  X  X
133      X  X  X
144  X,Y,Z  X  Y