Question

我试图找到嵌套列表最大值的索引。

我有三个清单[[1,2,3,4]，[5,6,7,8]，[9,10,11,12]]

我希望我的输出能给出值12的索引

输出结果如下：

最大元素的位置为（2,3）。

谢谢！

Answer 1

使用numpy的argmax，然后解开索引：

>>> L = [[1,2,3,4], [5,6,7,8], [9,10,11,12]]
>>> a = np.array(L)
>>> np.unravel_index(np.argmax(a), a.shape)
(2, 3)

Answer 2

如果你不想使用numpy，你可以像这样手动进行搜索：

a = [[1,2,3,4], [5,6,7,8], [9,10,11,12]]

maxEntry = None
coords = (None, None)
for x, sublist in enumerate(a):
    for y, entry in enumerate(sublist):
        if maxEntry == None or entry > maxEntry:
            maxEntry = entry
            coords = (x,y)

print(coords)

Answer 3

没有numpy的解决方案（假设列表及其第一个元素永远不会为空）：

def maxIndex(nestedList):
    m = (0,0)
    for i in range(len(nestedList)):
        for j in range(len(nestedList[i])):
            if nestedList[i][j] > nestedList[m[0]][m[1]]:
                m = (i,j)
    return m

Answer 4

mylist1 = [[1,2,3,4], [5,6,7,8], [9,10,11,12]]
mylist2 = [[1,1,1,1,9], [5,5,5,5,5]] 

def myfunc(mylist):
    xx =  mylist.index(max(mylist, key = lambda x:max(x)))
    yy =  mylist[xx].index(max(mylist[xx]))
    print(xx,yy)

myfunc(mylist1)
myfunc(mylist2)

返回

(2, 3)
(0, 4)

Answer 5

如果您想要一个适用于任何深度嵌套列表的解决方案，那么您可以试试这个。

def find_index_of_max(nested_lst):
    _max, idx_path, cpath = None, None, []

    def find_max(lst, depth = 0):
        nonlocal idx_path, _max
        if len(cpath) - 1 < depth:
            cpath.append(0)

        for i, val in enumerate(lst):
            cpath[depth] = i
            if _max is None or val > _max:
                idx_path = tuple(cpath)
                _max = val
            elif isinstance(val, list):
                find_max(val, depth + 1)

    find_max(nested_lst)
    return idx_path

#Output
>>> find_index_of_max([[1,2,3,4], [5,6,7,8], [9,10,11,12]])
(2, 3)
>>> find_index_of_max([[1,2,[16, [17, 18], 3], 3,4], [5,6,7,8], [9,10,11,12]])
(0, 2, 1, 1)

查找嵌套列表Python 3.x的最大值的索引

5 个答案: