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 s的情况下，上述解决方案将为您提供找到该值的最大索引。

如果这不是你想要的，你可以将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))

Answer 2

您可以执行以下操作：

max(data, key=lambda x: x[0])

或者，更有效的替代方案：

max(data, key=operator.itemgetter(0))

这是时间比较：

In [11]: timeit.timeit('max(data, key=lambda x: x[0])', setup='import operator, random; data=[[random.randint(1,100) for _ in range(100)] for _ in range(100)]')
Out[11]: 8.272005081176758

In [12]: timeit.timeit('max(data, key=operator.itemgetter(0))', setup='import operator, random; data=[[random.randint(1,100) for _ in range(100)] for _ in range(100)]')
Out[12]: 5.0041139125823975

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

2 个答案: