返回第一项或None

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

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

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

当前回答

一些人建议这样做:

list = get_list()
return list and list[0] or None

这在许多情况下都有效，但它只在列表[0]不等于0、False或空字符串时才有效。如果列表[0]为0、False或空字符串，该方法将错误地返回None。

我在自己的代码中创建了这个错误太多次了!

其他回答

最好的方法是:

a = get_list()
return a[0] if a else None

你也可以在一行中完成，但这对程序员来说很难读懂:

return (get_list()[:1] or [None])[0]

这个怎么样:

(my_list and my_list[0])或None

注意:这应该适用于对象列表，但它可能会返回错误的答案，在数字或字符串列表的情况下，每个下面的注释。

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

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

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

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

优点:

一线

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

-容易理解

try:
    return a[0]
except IndexError:
    return None

if mylist != []:

       print(mylist[0])

   else:

       print(None)

推荐文章