Skip to main content
 首页 » 编程设计

python之如何确定一列的元素是否在另一列中

2025年05月04日15Free-Thinker

我有一个名为 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