为什么numpy.where给我这个输出？

Question

我正在关注numpy.where的文档页面，并找到以下代码：

'A', 'B', 'C' or 'D'

我不明白为什么np.where（x＆gt; 5）给出了提到的输出。如果之前有人询问过，我很抱歉，但我没有找到任何相关问题。请帮忙。

Answer 1

numpy.where返回条件为True的索引。因此，在您的示例x > 5中，True位于以下索引

  [(2,0), (2,1), (2,2)]
#  ^6.    ^7.    ^8.

如果要从原始数组中提取这些元素，这可能很有用，例如

>>> x[np.where( x > 5 )]
array([6., 7., 8.])