新しいデータフレーム列のパンダ条件付き作成

Question

この質問はPandas conditional creation of a series/dataframe columnの拡張です。私たちは、このデータフレームがあった場合 は：

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} 

どのように私はそれを行うだろうか？

Answer 1

あなたはisinと最後fillnaでlocを使用することができます。

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 

Answer 2

この使用np.whereをお試しください：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