返回第一项或None

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

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

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

当前回答

不是python中等价于c风格的三元运算符

cond and true_expr or false_expr

ie.

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

其他回答

我的用例只是设置一个局部变量的值。

就我个人而言，我发现try and except风格更容易阅读

items = [10, 20]
try: first_item = items[0]
except IndexError: first_item = None
print first_item

而不是切片列表。

items = [10, 20]
first_item = (items[:1] or [None, ])[0]
print first_item

下面的代码涵盖了使用lambda的几个场景:

l1 = [1,2,3]
l2 = []
l3 = None
first_elem = lambda x: x[0] if x else None
print(first_elem(l1))
print(first_elem(l2))
print(first_elem(l3))

这个怎么样:

(my_list and my_list[0])或None

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

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

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

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

if mylist != []:

       print(mylist[0])

   else:

       print(None)

推荐文章