A&Bを同時に抽出する

Question

前提・実現したいこと

pandasで、データフレームの"A"と"B"が含まれている行を抽出したい
###ここに質問の内容を詳しく書いてください。
データフレームの中から、特定の組み合わせが入っている行を抽出するプログラムを書きたいです。

（例） df
ID　　age products
0 300 30.0 A,B
1 302 31.0 B
2 303 25.0 B,A
3 304 29.0 C,A
4 305 32.0 B,A

上記から、productsに、AとBを同時に持っている行を抽出したいです。
ID　　age products
0 300 30.0 A,B
2 303 25.0 B,A
4 305 32.0 B,A

###試したこと
df.query('products="A,B"')
しかしこれだと、A,Bの組み合わせのデータしか検索できませんでした。

Answer 1

productsとして'A,C,B,D'のように2個以上のものが入っている場合があるなら

>>> import pandas as pd
>>> df = pd.DataFrame({'ID':[300,302,303,304,305,306], 'age':[30.0,31.0,25.0,29.0,32.0,28.0], 'products': ['A,B','B','B,A','C,A','B,A','A,C,B,D']})
>>> df2 = df[df['products'].apply(lambda x: 'A' in x.split(',') and 'B' in x.split(','))]
>>> print(df2)
    ID   age products
0  300  30.0      A,B
2  303  25.0      B,A
4  305  32.0      B,A
5  306  28.0  A,C,B,D

Comments

[UedaJiro]

初歩的な質問に、とても親切にご回答頂き本当にありがとうございました！

Answer 2

product列の内容を集合に変換し、集合{'A','B'}と比較するとよいです。

import pandas as pd
from io import StringIO

s = """ID   age     products
300  30.0    A,B
302  31.0    B
303  25.0    B,A
304  29.0    C,A
305  32.0    B,A"""
df = pd.read_csv(StringIO(s), sep='\s+')

ab = df['products'].apply(lambda x: set(x.split(',')) == set(('A','B')))

print(df[ab])
#    ID   age products
#0  300  30.0      A,B
#2  303  25.0      B,A
#4  305  32.0      B,A

Comments

[UedaJiro]

丁寧にご回答して頂き本当にありがとうございました。
お蔭様で疑問を解決することができました。