np.where()[0] 和 np.where()[1]

最新推荐文章于 2023-11-08 14:45:01 发布

原创最新推荐文章于 2023-11-08 14:45:01 发布 · 2.1k 阅读

14 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#numpy #python #索引

收录于

Python 专栏收录该内容

5 篇文章

订阅专栏

本文通过实例演示了如何使用NumPy的where函数来获取数组中符合条件元素的位置，并展示了如何利用这些位置索引来提取特定元素。这有助于理解NumPy中数组索引的工作原理。

import numpy as np

a = np.arange(12).reshape(3,4)
print('a:', a)
print('np.where(a > 5):', np.where(a > 5))
print('a[np.where(a > 5)]:', a[np.where(a > 5)])
print('np.where(a > 5)[0]:', np.where(a > 5)[0])
print('np.where(a > 5)[1]:', np.where(a > 5)[1])
print(a[np.where(a > 5)[0], np.where(a > 5)[1]])

a: [[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
np.where(a > 5): (array([1, 1, 2, 2, 2, 2]), array([2, 3, 0, 1, 2, 3]))
a[np.where(a > 5)]: [ 6  7  8  9 10 11]
np.where(a > 5)[0]: [1 1 2 2 2 2]
np.where(a > 5)[1]: [2 3 0 1 2 3]
[ 6  7  8  9 10 11]

np.where()[0] 表示行索引，np.where()[1]表示列索引