[Python] Boolean Or "Mask" Index Arrays filter with numpy

最新推荐文章于 2025-08-08 11:09:55 发布

转载最新推荐文章于 2025-08-08 11:09:55 发布 · 54 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Answer1215/p/8067472.html

文章标签：

#python

本文介绍使用Python的NumPy库进行数组元素过滤与替换的方法。通过计算平均值，并利用布尔数组来选择低于平均值的元素进行替换，展示了NumPy在数组操作上的灵活性和效率。

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

NumPy Reference: Indexing

Integer array indexing
Boolean array indexing

Note: The expression a < mean produces a boolean array, like:

[[False, False, True, False, False, False, True, True, True],
 [True, True, False, False, True, True, False, True, True]]

def filter():
    a = np.array([
        (20,20,10,23,26,32,10,5,0),
        (0,2,50,20,0,1,28,5,0)
    ])

    mean = a.mean()
    print("mean",mean) #14.0
    print(a[a<mean]) #[10 10  5  0  0  2  0  1  5  0]

    # replace all which value which is less than mean value
    a[a<mean]=mean
    print(a)
    """
    [[20 20 14 23 26 32 14 14 14]
    [14 14 50 20 14 14 28 14 14]]
    """


if __name__ == "__main__":

    filter()