python - 列表列表及其索引中的最大值

Question

li = [[1,2], [2,3], [7,6]]

如何有效地找到最大值及其索引？假设对于 li 我想要:

max_value = 7

max_index = (2, 0)

我可以这样做:

max_value = 0
for row_idx, row in enumerate(alignment_matrix):    
    for col_idx, col in enumerate(row):
        if col > max_value:
            max_value = col
            max_index = (row_idx, col_idx)

但我需要一种不使用太多不必要变量的有效方法。

Answer 1

使用 max和 generator expression ，你可以更简短地表达它:

max_value, max_index = max((x, (i, j))
                           for i, row in enumerate(li)
                           for j, x in enumerate(row))

但是，时间复杂度是相同的，因为这个也使用了嵌套循环。

更新

正如@jonrsharpe 指出的，在重复 max_value 的情况下，上述解决方案将为您提供找到该值的最大索引。

如果这不是您想要的，您可以将 key 函数参数传递给 max 以自定义行为:

max_value, max_index = max(((x, (i, j))
                            for i, row in enumerate(li)
                            for j, x in enumerate(row)),
                           key=lambda (x, (i, j)): (x, -i, -j))