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

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])