Python如果其他方法返回多个值，DataFrame值

Question

我想做以下几点：

1. 传递Dataframe['ID‘以发挥作用
2. 如果球队是蓝队，那就把“Y”改为“他会打吗？”并将“Blue”值返回到“为什么”。

下面是我对代码的尝试。

def play(df):
if df['Team'] in list(['Blue']):
    return 'Exclude','**************'  

df['Can he play?'],df['Why?'] = df.apply(play, axis = 1)

我不知道如何从条件语句返回dataframe值。

如何返回“Blue”(df‘’Team‘中的值)

​

​

Answer 1

这可以分为两个步骤，如下所示：

df = pd.DataFrame({'Team': ['Blue', 'Green', 'Blue', 'Red']})
colour = 'Blue'
df['Can he play?'] = np.where(df['Team'] == colour, 'Y', None)
df['Why?'] = np.where(df['Team'] == colour, colour, None)

    Team Can he play?  Why?
0   Blue            Y  Blue
1  Green         None  None
2   Blue            Y  Blue
3    Red         None  None