python - pandas 数据框中的 If/Else 流控制

Question

我是Python新手，正在研究逻辑语句。 我的目标是统计各队的进球得分。 (即如果一支球队进球，我将分配 1，而对手将分配 -1)。下面是数据快照。 Data is below

我写的逻辑语句如下:

if data['team']== data['hometeam_team1']:
   data['run_score'] = 1
else:
    data['run_score'] = -1

但它给我带来了值(value)错误:

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

有人可以建议吗？ 非常感谢您的帮助。 谢谢 昂

Answer 1

使用panda的好处是矢量化计算。换句话说，您很少需要使用显式的 for 循环或 if/else 子句来执行计算在每一行。

相反，您可以对 pd.Series 对象执行计算。在此示例中，一种有效的解决方案是使用 numpy.where它的作用类似于矢量化的 if/else 子句:

import numpy as np

data['run_score'] = np.where(data['team']== data['hometeam_team1'], 1, -1)