Pythonic方式允许数据,列表中存在其他无

时间:2016-04-13 09:30:07

标签: python

假设我有一个数据列表

  

[“first”,“second”,“third”]

我有一个简单的函数,它接收参数

def simple_function(arg1)

我想给出一个参数,检查列表中是否存在这个参数并在之后使用它。在其他情况下 - 将其设置为无。像这样的东西

def simple_function(arg1):
    l = ["first", "second", "third"]
    if arg1 not in l:
        arg1 = None

最狡猾的方式是什么?

1 个答案:

答案 0 :(得分:3)

你可以这样做:

arg1 = None if arg1 not in ["first", "second", "third"] else arg1