python - 根据条件用数据填充 NaN

Question

我的代码是这样的:

if df['FLAG'] == 1:
    df['VAL'] = df['VAL'].fillna(median)
elif df['FLAG'] == 0:
    df['VAL'] = df['VAL'].fillna(0)

返回 - DataFrame 的真值不明确。使用 a.empty、a.bool()、a.item()、a.any() 或 a.all()。

我试过像蒙版一样做，然后用 a.all() 应用它，但没有成功。非常感谢您的启发!

编辑:我的问题就在这里 - Link

Answer 1

这是一个逐元素操作，您可以对其进行向量化。使用 np.where 构建一个数组并将其传递给 fillna。

df['VAL'] = df['VAL'].fillna(np.where(df['FLAG'], median, 0))