熊猫条件创建新的数据帧列

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

您可以使用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 

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