import numpy as np
t = np.arange(15).reshape(3,5).astype(np.float)
t[1,2:] = np.nan
print(t)
#將每列的nan改成每列的平均值
for i in range(t.shape[1]):
temp_col = t[:,i]
#判斷是否存在nan
nan_num = np.count_nonzero(temp_col != temp_col)
if nan_num:
#取出每列中不爲nan的值
no_nan_array = temp_col[temp_col == temp_col]
#給nan賦值
temp_col[np.isnan(temp_col)] = no_nan_array.mean()
print(t)