Numpy将nan、inf替换为每列均值或者0

小白炼丹师

于 2020-10-13 16:20:17 发布

阅读量6.8k

点赞数 4

分类专栏：笔记文章标签： python numpy

本文链接：https://blog.youkuaiyun.com/aaron_zy7/article/details/109054073

版权

这篇博客介绍了如何在Numpy中处理nan和inf值，提供将nan替换为每列均值以及将inf替换为0的方法。作者分享了参考链接，并幽默地自称为‘缝合大师’。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

nan和inf替换成0

import numpy as np
a = np.array([[np.nan, np.nan, 1, 2], [np.inf, np.inf, 3, 4], [1, 1, 1, 1], [2, 2, 2, 2]])
print (a)
where_are_nan = np.isnan(a)
where_are_inf = np.isinf(a)
#nan替换成0,inf替换成nan
a[where_are_nan] = 0
a[where_are_inf] = 0
print(a)

参考：https://blog.youkuaiyun.com/helei001/article/details/53066317

nan值替换成每列的均值

# coding=utf-8
import numpy as np
 
 
def fill_ndarray(t1):
    for i in range(t1.shape[1]):  # 遍历每一列（每一列中的nan替换成该列的均值）
        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_