python - 使用 lambda 在 Pandas 中创建一个新变量

Question

如果我想在 Pandas 中执行以下操作，我该如何使用 Lambda 执行此操作:

df['c'] = 0
df['c'][(df['a']==1) | (df['b'] ==1)] = 1

谢谢

Answer 1

我无法证明使用 lambda

df.assign(c=df[['a', 'b']].eq(1).any(1).astype(int))

但如果你坚持:

df.assign(c=lambda d: d[['a', 'b']].eq(1).any(1).astype(int))