python - 如何在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]])

Answer 2

您可以使用嵌套循环执行此操作：

indices = []
for i, row in enumerate(grid):
    for j, val in enumerate(row):
        if val > threshold:
            indices.append((i, j))

但很可能您应该使用numpy，它提供高速矢量化操作和非常方便的语法。使用 numpy ndarray 而不是嵌套列表，代码如下：

numpy.where(grid > threshold)

Answer 3

第一种方法是，没有任何库：

x = [ [100, 200, 100, 250],
      [125, 100, 250, 100] ]
indices = []
for i, range in enumerate(x):
    for j, elem in enumerate(range):
         if elem >= 200:
             indices.append((i, j))

结果是：

>>> indices
[(0, 1), (0, 3), (1, 2)]