Python 从其他两列中删除列值

我有以下数据帧

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

---

我想创建

必需列

，这样如果值在

条件

列中，则应选择

团队

和

团队2

列中的

条件

以外的值。在熊猫身上这样做的最佳方式是什么？

使用

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

---

使用

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

您可以使用：

输出：

        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

您可以使用：

输出：

        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