查找列表中具有第二高元素的所有数组

Question

假设我在 Python 3.2 中有一个数组列表，并且我想输出一个包含 每个 数组元素的数组，以及它们在列表中的索引位置，这些元素具有最高第二个元素。我怎样才能以最可扩展的方式实现这个目标（即，不必使用嵌套的 for-loop ）？

输入

a = [[2,3], [1,4,5], [1,4,6,2], [3,3,5], [9,4]]

预期输出

res = [[[1,4,5], 1], [[1, 4, 6,2], 2], [[9,4], 4]]

有人可以帮助我在不使用嵌套for-loop 的情况下如何做到这一点吗？

Answer 1

你可以这样做：

b = max(a, key=lambda x:x[1])[1]  
[[j, i] for i, j in enumerate(a) if j[1]==b]
Out[6]: [[[1, 4, 5], 1], [[1, 4, 6, 2], 2], [[9, 4], 4]]