np.where返回满足内部条件的数列a中的索引值。
np.where(condition)[0]返回一个numpy数列。
a=np.array([3,4,5,7,8])
np.where(a>7)[0]
#output:array([4])
a=np.array([3,4,5,7,8])
b=np.array([1,2,3,4,5])
np.where((a>=5)&(b>=4))
#output:(array([3, 4]),)
本文介绍如何利用NumPy的np.where函数筛选数组中满足特定条件的元素,并获取其索引位置。通过实例演示了单条件及复合条件的筛选过程。
np.where返回满足内部条件的数列a中的索引值。
np.where(condition)[0]返回一个numpy数列。
a=np.array([3,4,5,7,8])
np.where(a>7)[0]
#output:array([4])
a=np.array([3,4,5,7,8])
b=np.array([1,2,3,4,5])
np.where((a>=5)&(b>=4))
#output:(array([3, 4]),)
3601
218
986
3958