如何在嵌套列表Python中定位索引

Question

从输入导入列表

def find_peak(m: List[List[int]]) -> List[int]:
    """
    Given an non-empty elevation map m, returns the cell of the
    highest point in m.

Examples (note some spacing has been added for human readablity)
>>> m = [[1,2,3],
         [9,8,7],
         [5,4,6]]
>>> find_peak(m)
[1,0]
>>> m = [[6,2,3],
         [1,8,7],
         [5,4,9]]
>>> find_peak(m)
[2,2]
"""
max_location = []
for sublist in m:
    max_location.append(max(sublist))
max_location = max(max_location)
for sublist in m:
    if max_location in sublist:
        return (m.index(max_location),sublist.index(max_location))

这实际上不起作用，因为它只返回不在列表中的数字

Answer 1

您还可以充分利用enumerate。首先查找具有最大数目的行（及其索引）。然后在第二行找到那个数。在

在这两种情况下，您都需要为max函数提供一个键，以便它考虑值（而不是索引）：

def find_peak(m):
    i, max_row = max(enumerate(m), key=lambda x: max(x[1]))
    j, max_val = max(enumerate(max_row), key=lambda x: x[1])
    return [i, j]

输出

^{pr2}$