Skip to main content
 首页 » 编程设计

python之使用 apply 在 pandas 数据框中的其他列中查找列值

2025年05月04日25zhoujg

我有一个具有以下形式的 df:

        id  Tg n0  n1  n2 
fddef72025   W  A   B   C 
fddef72025   A  A   B   C 
fddef72025   Y  A   B   C 
fddef72025   B  A   B   C 

如果 Tg 在 df.n0、df.n1 或 中,我需要创建一个新列并存储 1 >df.n2,例如上面的df:

        id  Tg n0  n1  n2  exists 
fddef72025   W  A   B   C       0 
fddef72025   A  A   B   C       1 
fddef72025   Y  A   B   C       0 
fddef72025   B  A   B   C       1 

到目前为止,我的解决方案涉及“iterrows”,但我确定有办法df.apply这个?

请您参考如下方法:

国际联合会

df.loc[:,'n0':].isin(df.Tg).any(1).astype(int) 
0    0 
1    1 
2    0 
3    1 
dtype: int64