[Python知识库]数据分析：numpy——练习

import numpy as np
def fill_ndarray(t1):
    for i in range(t1.shape[1]):# 遍历每一列
        temp_col = t1[:,i]# 当前的一列
        nan_num = np.count_nonzero(temp_col != temp_col)
        if nan_num != 0:# 不为0，说明当前这一列中有nan
            temp_not_nan_col = temp_col[temp_col == temp_col]# 当前有一列不为nan的array
            temp_col[np.isnan(temp_col)] = temp_not_nan_col.mean()# 选中当前为nan的位置，把值赋值为不为nan的均值
    return t1
t1 = np.arange(12).reshape(3, 4).astype(float)
t1[1, 2:] = np.nan
print(t1)
t1 = fill_ndarray(t1)
print(t1)