添加一列，该列的值可以是多个不同的值

Question

我正在尝试向pandas数据集中添加一个新列，该数据集从一列中提取数字月份，并将相应的季节存储到新列中。你知道吗

所以本质上

if month == 12, season = 'Winter'
elif month == 2, season = 'Spring'
elif month == 5, season = 'Summer'
else month == 8, season = 'Fall'

我还没有看到一个真正明确的解决办法。我已经了解了如何使用2个条件值，但对pandas和python还是相当陌生的。你知道吗

编辑：我能够让它与下面列出的解决方案之一工作（谢谢！），但我应该提到，我还需要包括1、3、4、6、7、9、10、11个月

Answer 1

你能试试下面的吗。公平的警告，因为没有输入和预期输出的样本，所以没有测试它。你知道吗

import pandas as pd
import numpy as np
conditions=[df['month']==12,
            df['month']==2,
            df['month']==5,
            df['month']==8]
choices=['Winter','spring','Summer','fall']

df['season']=np.select(conditions,choices,default=np.nan)