我想要一个函数返回N列表中最大的一个。列表中有两个项目我可以写:
l1 = [3, 4, 5]
l2 = [4, 5, 6, 7]
def f(L):
if(len(L[0]) > len(L[1])):
return L[0]
else:
return L[1]
我使用f([l1, l2])
运行。
然而,随着更多的列表,它变成了if语句的连续性,而且很难看。
您如何非常有效地返回N个列表中最大的一个?
答案 0 :(得分:28)
将max
与key=len
一起使用。
In [3]: max([l1, l2], key=len)
Out[3]: [4, 5, 6, 7]
这将检索列表列表的(第一个)最长列表。
实际上,这也适用于字符串(以及具有len属性的其他对象)。
In [4]: max(['abcd', 'ab'], key=len)
Out[4]: 'abcd'
In [5]: max([(1, 2), (1, 2, 3), (1,)], key=len)
Out[5]: (1, 2, 3)
In [6]: max(['abc', [1, 2, 3]], key=len)
Out[6]: 'abc'
注意:我们也可以将这些项作为参数传递:
In [7]: max(l1, l2, key=len)
Out[7]: [4, 5, 6, 7]
max
读取:从key
的角度查看时(如果您通过key
)获取列表中最大的项目。
它大致等效于以下代码*(在python 3中),但实际来源是written in C(效率更高,实际测试的更多,所以请继续使用max而不是此):
def my_max(*L, key=None): # in python 2 we'd need to grab from kwargs (and raise type error if rogue keywords are passed)
L = list(L[0]) if len(L) == 1 else L # max takes iterable as first argument, or two or more arguments...
if not L:
raise ValueError("my_max() arg is an empty sequence")
if key is None: # if you don't pass a key use the identity
key = lambda x: x
max_item, max_size = L[0], key(L[0])
for item in L[1:]:
if key(item) > max_size:
max_item, max_size = item, key(item)
return max_item
*我把它作为练习使用迭代器而不是列表来编写...并修复任何其他错误!