Панды условное создание нового столбца фрейма данных

Этот вопрос является продолжением условного создания пандами ряда/столбца фрейма данных. Если бы у нас был такой фрейм данных: 

    Col1       Col2
1    A          Z
2    B          Z           
3    B          X
4    C          Y
5    C          W

И мы хотели сделать эквивалент: 

if Col2 in ('Z','X') then Col3 = 'J' 
else if Col2 = 'Y' then Col3 = 'K'
else Col3 = {value of Col1}

Как я мог это сделать?

2 2
python pandas dataframe if-statement multiple-columns

2 ответа:

Вы можете использовать loc с помощью isin и последнее fillna:

df.loc[df.Col2.isin(['Z','X']), 'Col3'] = 'J'
df.loc[df.Col2 == 'Y', 'Col3'] = 'K'
df['Col3'] = df.Col3.fillna(df.Col1)
print (df)
  Col1 Col2 Col3
1    A    Z    J
2    B    Z    J
3    B    X    J
4    C    Y    K
5    C    W    C
3

Попробуйте использовать np.где: outcome = np.where(condition, true, false) 

  df["Col3"] = np.where(df['Col2'].isin(['Z','X']), "J", np.where(df['Col2'].isin(['Y']), 'K', df['Col1']))

  Col1 Col2 Col3
1    A    Z    J
2    B    Z    J
3    B    X    J
4    C    Y    K
5    C    W    C
0