python - 在数据帧列中填充键入 NA 的字符串时出错

Question

我有一个 pandas 数据框列，其中有浮点值和字符串类型 NA 值。我需要使用以下代码将这些 NA 替换为均值。

trainTestJoin["col1"] = trainTestJoin.groupby("col2")["col1"].
     transform(lambda x: x.fillna(x.median()))

我正在获取

TypeError: could not convert string to float: NA

我尝试在填充之前转换为。

trainTestJoin["LotFrontage"].astype(float)

但它给出了同样的问题。如何解决这个问题？

Answer 1

使用 pd.to_numeric 转换为数字它支持通过 errors='coerce' 将不可转换值转换为 float NaN:

df['col1'] = pd.to_numeric(df['col1'], errors='coerce')

然后直接使用groupby + transform:

df['col1'] = df['col1'].fillna(df.groupby('col2')['col1'].transform('mean'))