大熊猫从其他2列中删除列值

Question

我有以下数据框

team          team2     condition   Required_Col
Australia   Sri Lanka   Sri Lanka   Australia
Australia   Sri Lanka   Sri Lanka   Australia
Australia   Sri Lanka   Sri Lanka   Australia
Australia   Sri Lanka   Australia   Sri Lanka
Australia   Sri Lanka   Australia   Sri Lanka

我想创建Required_Col ，以便如果该值位于condition列中，则应选择team和team2列中除condition以外的其他值。 在熊猫中做到这一点的最佳方法是什么？

Answer 1

使用np.where

df['required'] = np.where(df.condition == df.team, df.team2, df.team)

        team     team2  condition   required
0  Australia  SriLanka   SriLanka  Australia
1  Australia  SriLanka   SriLanka  Australia
2  Australia  SriLanka   SriLanka  Australia
3  Australia  SriLanka  Australia   SriLanka
4  Australia  SriLanka  Australia   SriLanka

Answer 2

您可以使用.apply（） ：

df['Required_Col'] = df.apply(lambda x: x['team2'] if (x['condition']==x['team']) else x['team'], axis=1)

输出：

        team    team2       condition   Required_Col
0   Australia   SriLanka    SriLanka    Australia
1   Australia   SriLanka    SriLanka    Australia
2   Australia   SriLanka    SriLanka    Australia
3   Australia   SriLanka    Australia   SriLanka
4   Australia   SriLanka    Australia   SriLanka