python - 遍历 pandas dataframe 列并根据条件创建另一列

Question

我有一个数据框 df

ID  ID2 escto1 escto2   escto3
1   A   1   0   0
2   B   0   1   0
3   C   0   0   3
4   D   0   2   0

所以要么使用索引要么使用通配符

like column name 'escto*'
if df.iloc[:, 2:]>0 then df.helper=1

或

df.loc[(df.iloc[:, 3:]>0,'Transfer')]=1

这样输出就变成了

ID  ID2 escto1  escto2  escto3  helper
1   A   1   0   0   1
2   B   0   1   0   1
3   C   0   0   3   1
4   D   0   2   0   1

Output

Answer 1

一种选择是使用 bool 输出:

df.assign(helper = df.filter(like='escto').gt(0).any(1).astype(int))

   ID ID2  escto1  escto2  escto3  helper
0   1   A       1       0       0       1
1   2   B       0       1       0       1
2   3   C       0       0       3       1
3   4   D       0       2       0       1