我有一个包含“预算”和“类型”属性的电影数据集。

我想用每种类型的平均预算来填补预算的缺失值。

我首先创建两个有或没有预算的数据框。

BudgetNull = data[data['budget'].isnull()]
BudgetNotNull = data[data['budget'].notnull()]

然后，根据 BudgetNotNull 数据集计算每种类型的平均预算。

budget_of_genre = BudgetNotNull.groupby('genres')['budget'].mean()

最后，我想根据它的类型填写 BudgetNull 的预算。

例如，“海洋男孩”是一部动作片，因此，填写budget_of_genre['Action']。

我如何通过 for 循环做到这一点？还是有其他方法？