python - 如果月份 = 十月、十一月或十二月，则从年份中减去 -1，否则年份 python

Question

如果该行的“月份”列中的月份 ==“十月”、“十一月”、“十二月”，我将尝试从数据框中的“yy”列中减去 (1) 年。

我尝试了很多方法，但都陷入了困境。下面的循环使整个 yy2 列成为 yy-1，而不仅仅是月份列中包含 10 月、11 月或 12 月的行。

yy2 = []

for row in df['month']:
    if row != 'October':
        yy2 = df['yy']
    else:
        yy2 = (df['yy'])-1

df['yy2'] = yy2

Answer 1

使用 Series.where

month_selection = { 'October', 'November', 'December'}
df['yy2'] = df['yy'].where(~df['month'].isin(month_selection ), df['yy'] - 1)

更简单的实现

df['yy2'] = df['yy'].copy()
sel = df['month'].isin(month_selection )
df.loc[sel, 'yy2'] = df.loc[sel, 'yy'] - 1