我有一个名为 df 的 DataFrame,我想知道列 item 是否包含元素 a 和 c.
import pandas as pd
df=pd.DataFrame({'id':[1,2,3],'item':['a,b,c','c,d,e','a,b']})
输入:
df
id item
0 1 a,b,c
1 2 c,d,e
2 3 a,b
在新列 contain_a 中,1 表示列 item 包含元素 a ,即与 contain_c 列相同。
预期:
df
id item contain_a contain_c
0 1 a,b,c 1 1
1 2 c,d,e 0 1
2 3 a,b 1 0
请您参考如下方法:
对于执行此操作的通用方法,可以使用 str.get_dummies指定分隔符
df.item.str.get_dummies(',').add_prefix('contains_')
contains_a contains_b contains_c contains_d contains_e
0 1 1 1 0 0
1 0 0 1 1 1
2 1 1 0 0 0
