Pandas DataFrame：简单赋值 - 两列值之差到另一列值的下限，循环不起作用

Question

您将如何在 Pandas 中执行以下操作？

import math
for index, row in data.iterrows():
  if row["year"] == 0:
    row["year"] = math.floor((row["death"] - row["birth"])/2)

这个循环不起作用，但我想做的是，如果列年的值为 0，则将差异的下限除以死亡和出生列的 2 分配给列年。我知道你应该避免在 Pandas 中循环这可能有一个简单的解决方案，但我现在无法弄清楚。

Answer 1

使用numpy.where ：

import numpy as np

df['newcol'] = np.where(df['year'] == 0, math.floor((df['death'] - df['birth'])/2), df['year'])

这本质上是：

np.where(condition, if True then, if False then)

Answer 2

切片loc ：

df.loc[df['year'] == 0, 'year'] = np.floor((df.loc[df['year'] == 0, 'death'] - df.loc[df['year'] == 0, 'birth']) / 2)

也许是一个更具可读性的解决方案：

mask = df['year'] == 0
df.loc[mask, 'year'] = np.floor((df.loc[mask, 'death'] - df.loc[mask, 'birth']) / 2)