python - 条件搜索返回一维数组的数组元组？

Question

我阅读了关于 numpy's where 的文档，我不明白如果我检查一个简单的一维数组的条件，为什么 where 会返回一个嵌套数组。

> import numpy as np
> my_array = np.random.randint(1,10, (20))
> np.where(my_array > 5)

(array([ 0,  1,  4,  6,  7,  8, 10, 11, 13, 15, 16, 17, 18]),)

为什么 np.where 在这种情况下返回一个元组？为什么嵌套结果？

Answer 1

我认为这是为了保持一致性，考虑一个二维数组:

import numpy as np
my_array = np.random.randint(1,10, (4, 5))
pos = np.where(my_array > 5)
my_array[pos]

您可以使用元组作为索引来选择位置中的所有值。