numpy过滤

我的辉

于 2021-07-06 16:02:31 发布

阅读量814

点赞数

CC 4.0 BY-SA版权

分类专栏： numpy

本文链接：https://blog.youkuaiyun.com/huihui__huihui/article/details/118522839

numpy 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了如何使用numpy的where函数进行条件索引，选取数组中大于3的元素，并演示了如何将这些元素替换为10。通过实例展示了numpy数组操作的基本技巧。

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

numpy过滤
np.where 返回的是索引

import numpy as np

x = np.arange(9.).reshape(3,  3)
/*

[[0. 1. 2.]
 [3. 4. 5.]
 [6. 7. 8.]]

*/
y = np.where(x >  3)

/*
(array([1, 1, 2, 2, 2]), array([1, 2, 0, 1, 2]))

（1 1） （1 2） （2 0） （2 1） （2 2）这些位置的值都大于3

*/


#############输出大于3
print (x[y])
######记得在y的前面加上[] 不然会报错

修改numpy里面的数字

import numpy as np

x = np.arange(9.).reshape(3,  3)

#############让大于3的元素都等于10
x[x>3]=10

print(x)