首页 > 解决方案 > 熊猫从其他 2 列中删除列值

问题描述

我有以下数据框

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这样的值,如果值在列中,那么应该选择conditionconditioninteam和column 之外的值。team2在 pandas 中执行此操作的最佳方法是什么?

标签: pythonpython-3.xpandasdataframe

解决方案


使用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

推荐阅读