使用 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