6

我想要一个函数返回 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 个列表中最大的一个?

4

1 回答 1

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阅读:当(如果你通过keykey.
大致相当于以下代码*(在 python 3 中),但实际源代码是用 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

*我把它作为练习使用迭代器而不是列表来编写......并修复任何其他错误!

于 2013-05-23T19:58:20.873 回答