所以我基本上有一个2和1/2维矩阵,我想从列表中的列表中返回一个元素。
def somefunc(x):
# What I want to do is return the max element within a matrix based on l[:][1]
return j
# example
l = [[[1,2,3],4],
[[5,6,7],8],
[[9,1,2],3]]
>>>somefunc(l)
[[5,6,7],8]
答案 0 :(得分:3)
在key
函数中使用max()
参数。
def somefunc(x):
return max(x, key=lambda e: e[1])