訪問 numpy.where 索引中的元素

Question

我是 Python 世界的初學者，希望有人能回答我的問題。 我有一個數組，需要訪問元素的某些索引，如下所示

x = np.random.rand(10)

x
array([ 0.56807058,  0.8404783 ,  0.86835717,  0.76030882,  0.40242679,
        0.22941009,  0.56842643,  0.94541468,  0.92813747,  0.95980955])

indx = np.where(x < 0.5)

indx
(array([4, 5], dtype=int64),)

但是，當我嘗試使用indx[0]訪問第一個元素時，它返回array([4, 5], dtype=int64) 。 我想要做的是訪問indx元素 4 和 5 。 感謝您研究我的問題和任何支持。

Answer 1

np.where返回一個索引tuple 。 在這種情況下，元組只包含一個索引數組。 這與where處理多維數組的方式一致。 它返回一個包含多個數組的元組，這些數組共同定義了非零元素的索引。

要從indx訪問4 ，您可以這樣做： indx[0][0] 。 第一個[0]選擇indx元組的第一個元素，即array([4, 5], dtype=int64) ，第二個訪問該數組的一個元素。