使用 lambda 如果基于 Pandas dataframe 中另一列的值的列的条件

Question

让我们考虑这样的 dataframe：

(index)    condition    number
0          'increase'   1
1          'do-nothing' 1

如果同一行中的condition元素等于“增加”，我想增加number列中的元素。 原则上，我想到了类似的东西：

# non-working example
df['number'] = df['number'].apply(lambda x: x+=1 if x['condition'] == 'increase' else x)

这个例子当然行不通。 有没有有效的方法来做到这一点？

Answer 1

我想你正在寻找：

df.loc[df['condition'] == 'increase', 'number'] += 1