将numpy矩阵中的NAN值用对应列上的平均值替换

原创于 2019-07-09 10:10:48 发布 · 2.6k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#numpy #nan

Python 专栏收录该内容

6 篇文章

订阅专栏

博客涉及Numpy和NaN相关信息技术内容，但具体信息缺失。Numpy是重要的Python库，NaN在数据处理中有特殊意义。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import pandas as pd
import numpy as np

a = np.array([[1,0,np.NAN],[2,np.NAN,3],[np.NAN,7,8]])

#第一步，获取每列平均值
col_mean = np.nanmean(a,axis=0)

#第二步，获取NAN值所在索引
inds = np.where(np.isnan(a)) #如果报错，可以考虑替换使用下面一行代码
#inds = np.where(pd.isnull(a))

#第三步，替换值
a[inds] = np.take(col_mean,inds[1])