numpy高级函数：where与extract

Posted 2022-12-18 shierlou-123

numpy高级函数：where与extract

1、numpy.where()函数，此函数返回数组中满足某个条件的元素的索引：

import numpy as np

x = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("x")
print(x)
y=np.where(x>5)
print(y)
print(x[y])

2、numpy.extract()函数，和where函数有一点相，不过extract函数是返回满足条件的元素，而不是元素索引，下面我们的例子返回数组x 的元素 模2等于0的元素。

condition=np.mod(x,2)==0
print(condition)
print(np.extract(condition,x))