如何在python中的列表矩阵中选择最大的数字？

Question

我有一个 list-of-list-of-lists，其中前两个充当“矩阵”，我可以在其中访问第三个列表

list3 = m[x][y]   

第三个列表包含字符串和数字的混合，但每个列表具有相同的大小和结构。让我们将此列表中的特定条目称为“感兴趣的数量”。此数字在此列表中始终具有相同的索引！

在 Python 中获取“兴趣”数量最多的列表的“坐标”(x,y) 的最快方法是什么？

谢谢！

（真的，我正在尝试选择 m[x][y][k] 中 k 是固定的最大数字，对于所有 x 和 y，并且“知道”它的地址是什么）

Answer 1

max((cell[k], x, y)
    for (y, row) in enumerate(m)
    for (x, cell) in enumerate(row))[1:]

此外，您可以将结果直接分配给几个变量：

(_, x, y) = max((cell[k], x, y)
                for (y, row) in enumerate(m)
                for (x, cell) in enumerate(row))

这是 O(n²)，顺便说一句。

Answer 2

import itertools

indexes = itertools.product( xrange(len(m)), xrange(len(m[0]))
print max(indexes, key = lambda x: m[x[0]][x[1]][k])

或使用 numpy

import numpy
data = numpy.array(m)
print numpy.argmax(m[:,:,k])

如果你有兴趣在 python 中加速操作，你真的需要看看 numpy。

Answer 3

假设“感兴趣的数量”在列表中的已知位置，并且会有一个非零最大值，

maxCoords = [-1, -1]
maxNumOfInterest = -1
rowIndex = 0
for row in m:
    colIndex = 0
    for entry in row:
        if entry[indexOfNum] > maxNumOfInterest:
            maxNumOfInterest = entry[indexOfNum]
            maxCoords = [rowIndex,colIndex]
        colIndex += 1
    rowIndex += 1

是一个简单的方法，它将在矩阵的大小上为 O(n²)。由于您必须检查每个元素，因此这是最快的解决方案。

@Marcelo 的方法更丰富，但可能不太可读。