numpy学习小结

NumPy（Numerical Python）是Python语言的一个扩充程序库。支持高级大量的维度数组与矩阵运算，此外也针对数组运算提供大量的数学函数库。

nan不能修改为0，一般改为平均数。

import numpy as np
# t1 = np.loadtxt('1.csv', delimiter=',', dtype=int)
# print(t1.shape)
# print(t1)
# print(type(t1))
# print(t1[[2, 4, 6]])
# print(t1.clip(1000, 2000))


def fill_array(nan_t):
    for i in range(nan_t.shape[1]):
        temp_col = nan_t[:, i]
        nan_num = np.count_nonzero(temp_col != temp_col)
        if nan_num != 0:
            temp_not_nan_col = temp_col[temp_col == temp_col]
            temp_col[np.isnan(temp_col)] = temp_not_nan_col.mean()

    return nan_t


if __name__ == '__main__':
    t1 = np.arange(24).reshape((4, 6)).astype("float")
    t1[1, 2:] = np.nan
    print(t1)
    print('*' * 60)
    t = fill_array(t1)
    print(t)