返回第一项或None

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

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

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

当前回答

最python惯用的方法是在迭代器上使用next()，因为list是可迭代的。就像@J.F。塞巴斯蒂安在2011年12月13日发表了这条评论。

next(iter(the_list)， None)如果the_list为空返回None。参见next() Python 2.6+

或者如果你确定the_list不是空的:

iter(the_list).next()参见iterator.next(

其他回答

最好的方法是:

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

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

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

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

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

最python惯用的方法是在迭代器上使用next()，因为list是可迭代的。就像@J.F。塞巴斯蒂安在2011年12月13日发表了这条评论。

next(iter(the_list)， None)如果the_list为空返回None。参见next() Python 2.6+

或者如果你确定the_list不是空的:

iter(the_list).next()参见iterator.next(

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

如果你发现自己试图从列表理解中提取第一件事(或None)，你可以切换到生成器来这样做:

next((x for x in blah if cond), None)

正方:如果blah是不可索引的，它是有效的，反面:它是不熟悉的语法。不过，当在ipython中破解和过滤东西时，它很有用。

推荐文章