应用函数不会替换数据框中的值

Question

我正在尝试根据一些相应组的中位数替换年龄的 nan 值。我使用 groupby 制作了一个名为 grouped_median 的表。这是我的代码：

def fillAges(row, grouped_median):
    return grouped_median.loc[row['Sex'], row['Class'], row['Title']]['Age'] 


df['Age'] = df.apply(lambda x : fillAges(x, grouped_median) if np.isnan(x['Age']) else x['Age'], axis=1)

df

如果我只打印这部分：

print(df.apply(lambda x : fillAges(x, grouped_median) if np.isnan(x['Age']) else x['Age'], axis=1))

这些值是正确的，但是一旦我查看 df，nan 值就不会被替换。我很感激任何帮助。谢谢！

编辑：正如 Nathaniel 所说，这段代码运行良好。实际上，df 是一个大数据框，它由训练数据集和测试数据集连接而成，并带有一个额外的标志列，即“训练”或“测试”。然后这就是我正在做的：

df[df['flag']=='train']['Age'] = df[df['flag']=='train'].apply(lambda x : fillAges(x, grouped_median) if np.isnan(x['Age']) else x['Age'], axis=1)

它不会工作。这会给我这个警告，但我认为这只是一个警告，并不意味着它实际上没有做任何事情！！！ “试图在 DataFrame 的切片副本上设置值。尝试使用 .loc[row_indexer,col_indexer] = value 代替请参阅文档中的警告：pandas.pydata.org/pandas-docs/stable/...

我需要做的就是从左侧删除 [df['flag']=='train'] 部分。

我仍然不确定为什么这种方法不起作用。如果有人有见识，我很感激知道。谢谢

Answer 1

您必须对函数进行格式化以获取系列并返回系列，而不是仅对系列中的一个元素进行操作。我在下面添加了函数series_op，它应该会为你做这件事。

def fillAges(row, grouped_median):
    return grouped_median.loc[row['Sex'], row['Class'], row['Title']]['Age'] 

def series_op(x):
    x['Age'] = fillAges(x, grouped_median) if np.isnan(x['Age']) else x['Age']
    return x


corrected_df = df.apply(series_op, axis=1)

我没有你的数据，也没有grouped_median，所以我无法复制你的问题。借助我编写的一些测试数据，我可以让它正常工作，但也能让你的数据正常工作。