返回第一项或None

我调用了一堆返回列表的方法。列表可能是空的。如果列表非空，我想返回第一项;否则，我想返回None。这段代码工作:

def main():
    my_list = get_list()
    if len(my_list) > 0:
        return my_list[0]
    return None

但是，在我看来，应该有一个简单的一行习语来做这件事。是吗?

当前回答

关于习语，有一个itertools食谱叫做nth。

来自itertools食谱:

def nth(iterable, n, default=None):
    "Returns the nth item or a default value"
    return next(islice(iterable, n, None), default)

如果你想要一行程序，可以考虑安装一个库来实现这个recipe，例如more_itertools:

import more_itertools as mit

mit.nth([3, 2, 1], 0)
# 3

mit.nth([], 0)                                             # default is `None`
# None

还有另一个只返回第一项的工具，叫做more_itertools.first。

mit.first([3, 2, 1])
# 3

mit.first([], default=None)
# None

这些itertools一般适用于任何可迭代对象，而不仅仅适用于列表。

其他回答

(get_list() or [None])[0]

这应该有用。

顺便说一句，我没有使用变量列表，因为那会覆盖内置列表()函数。

编辑:我之前有一个稍微简单，但错误的版本。

可能不是最快的解决方案，但没有人提到这个选项:

dict(enumerate(get_list())).get(0)

如果get_list()可以返回None，你可以使用:

dict(enumerate(get_list() or [])).get(0)

优点:

一线

-你只需要调用get_list()一次

-容易理解

Python 2 + 6。

next(iter(your_list), None)

如果your_list可以为None:

next(iter(your_list or []), None)

Python 2.4

def get_first(iterable, default=None):
    if iterable:
        for item in iterable:
            return item
    return default

例子:

x = get_first(get_first_list())
if x:
    ...
y = get_first(get_second_list())
if y:
    ...

另一个选择是内联上面的函数:

for x in get_first_list() or []:
    # process x
    break # process at most one item
for y in get_second_list() or []:
    # process y
    break

为了避免中断，你可以这样写:

for x in yield_first(get_first_list()):
    x # process x
for y in yield_first(get_second_list()):
    y # process y

地点:

def yield_first(iterable):
    for item in iterable or []:
        yield item
        return

你可以使用提取方法。换句话说，将代码提取到一个方法中，然后调用该方法。

我不会试图进一步压缩它，一行代码似乎比冗长的版本更难阅读。如果你使用提取方法，它是一行;)

坦率地说，我认为没有更好的成语了:your is clear and short -不需要什么“更好的”。也许吧，但这真的是一个品味的问题，你可以改变if len(list) > 0:与if list: -一个空列表将永远求值为False。

在相关的说明中，Python不是Perl(没有双关的意思!)，您不必获得尽可能最酷的代码。实际上，我在Python中见过的最糟糕的代码，也是非常酷的:-)，而且完全无法维护。

顺便说一下，我在这里看到的大多数解决方案都不考虑list[0]计算为False(例如空字符串或零)-在这种情况下，它们都返回None而不是正确的元素。

推荐文章