np.where函数使用方法

np.where函数使用方法：指定条件获取元素索引的函数

1.1、

a=np.array([10,12,9,3,19])
print(a)

[10 12 9 3 19]

a[a<10]

array([9, 3])

1.2

a=np.arange(0,20,2)
print(a)

[ 0  2  4  6  8 10 12 14 16 18]

np.where(a<10)

(array([0, 1, 2, 3, 4]),)

1.3

a=np.arange(12).reshape(3,4)
print(a)

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

np.where(a%2==0)

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

1.4

np.where(a%2==0,'even','odd')

array([['even', 'odd', 'even', 'odd'],
       ['even', 'odd', 'even', 'odd'],
       ['even', 'odd', 'even', 'odd']], dtype='<U4')

附件

关注公众号「水沐教育科技」，在手机上阅读所有教程，随时随地都能学习。内含一款搜索神器，免费下载全网书籍和视频。

公众号二维码
微信扫码关注公众号