如何在python多维列表中查找特定值以上的所有索引

Question

我在python中有一个带有一些浮点值的二维列表.我想找出列表中所有元素的索引,这些索引高于特定值.

例如,在下面的2 x 4列表中(为方便起见,整数):

100    200    100    250
125    100    250    100

我需要所有等于或大于200的值的索引.这些索引是[0] [1],[0] [3]和[1] [2].

请帮助解决这个问题.

谢谢.

Answer 1

Numpy非常方便:

>>> import numpy as np
>>> a = [[100, 200, 100, 250], [125, 100, 250, 100]]
>>> a=np.array(a)
>>> np.argwhere(a>=200)
      array([[0, 1],
             [0, 3],
             [1, 2]])