改进这段代码,只对浮点型的列进行填充# 使用均值填充缺失值 raw_data.fillna(value=means, inplace=True)
时间: 2024-05-07 22:23:42 浏览: 45
pandas 使用均值填充缺失值列的小技巧分享
下面是改进后的代码,使用了 `select_dtypes` 方法来仅选择浮点型的列:
```python
# 计算均值
means = raw_data.select_dtypes(include=['float']).mean()
# 使用均值填充缺失值
raw_data.fillna(value=means, inplace=True)
```
这样就只会对浮点型的列进行填充了。同时,使用了 `include` 参数来仅选择浮点型的列,避免了对整个数据集进行操作,提高了效率。
阅读全文