Присвоение значения переменной новому столбцу в пандах DataFrame

Question

Я создал 3 новые переменные a, b и c на основе определенных условий. Теперь я хочу снова присвоить эти переменные новому столбцу в DataFrame pandas, используя некоторые условия.

Мне нужен код, подобный тому, что я написал ниже, но более разумным способом. Приведенный ниже код работает нормально, но это не разумное решение. есть ли возможность написать цикл, возможно, итерационный цикл

df2['dest_col'] = np.where(df2['month'] == 'March-2016',a
                  ,(np.where(df2['month'] == 'April-2016',b
                  ,(np.where(df2['month'] == 'May-2016',c)))

Answer 1

Вы хотите numpy.select:

cond=[(df2['month'] == month) for month in df2['month'].unique()]
values=[a, b, c] 

df2['dest_col'] = np.select(cond,values)