瞭解有關numpy.where

Question

我讀了numpy.where(condition[, x, y])documentation，但我不明白的小例子：

>>> x = np.arange(9.).reshape(3, 3) 
>>> np.where(x > 5) 
Out: (array([2, 2, 2]), array([0, 1, 2])) 

一些人能夠解釋的結果是怎麼來？

Answer 1

第一陣列（array([2, 2, 2])）是行（array([0, 1, 2])）的索引，第二個是那些大於5

可以使用zip獲得值的精確指標值的列：

>>> zip(*np.where(x > 5)) 
[(2, 0), (2, 1), (2, 2)] 

或者使用np.dstack：

>>> np.dstack(np.where(x > 5)) 
array([[[2, 0], 
     [2, 1], 
     [2, 2]]]) 

Answer 2

它打印出的座標您康迪灰

import numpy as np 

x = np.arange(9.).reshape(3, 3) 
print x 
print np.where(x > 5) 

其中打印X打印：

[[ 0. 1. 2.] 
[ 3. 4. 5.] 
[ 6. 7. 8.]] 

和np.where(x > 5)打印所有元素的索引位置超過5

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

其中2,0 == 6和2,1 == 7和2,2 == 8