访问 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) ，第二个访问该数组的一个元素。