ndarry缺失值填充均值

#!usr/bin/env python
# -*- coding:utf-8 -*-
"""
@author: 86151
@file: page87.py
@time: 2021/04/10
@desc:
"""
import numpy as np


def fill_ndarray(t1):
    for i in range(t1.shape[1]): #遍历每一列
        tem_col=t1[:i] #当前的一列
        nan_num=np.count_nonzero(tem_col!=tem_col)
        if nan_num!=0: #不为0，说明当前这一列有nan
            temp_non_nan_col=tem_col[tem_col==tem_col] #当前一列不为nan的array
            #选中当前为nan的位置，把值赋为不为nan的均值
            tem_col[np.isnan(tem_col)]=temp_non_nan_col.mean()
    return t1

if __name__=="__main__":
    t1=np.arange(12).reshape(3,4).astype("float")
    t1[1,2:]=np.nan
    print(t1)
    t1=fill_ndarray(t1)
    print(t1)