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))
原文地址:https://www.cnblogs.com/shierlou-123/p/11620003.html
时间: 2024-10-11 01:54:36