将列表理解传递给 Python 中的函数

Question

我有一个函数，我想将列表理解作为输入传递给它。我收到关于 my_list 未定义的错误。我知道我可以将my_list 放在函数之外，但实际上该列表是作为函数的一部分生成的。

实际定义比较复杂，这里是简化的例子：

def my_def(list_comp):
    global my_list
    my_list = [[1,2],[3,4]]
    list_comp_output = list_comp
    return list_comp_output

print my_def([x[1] for x in my_list])

Answer 1

实际上 我们在 Python 中所拥有的只是运行时；没有单独的编译时间这样的东西¹。（在解释器的范围内）。并且函数也不例外。

所以你在这里所做的就是将my_list 定义为一个全局变量，并尝试在列表推导中使用它，而python 没有定义这样的东西。

您可以只运行您的函数 1 次，然后使用该列表理解：

def my_def(list_comp):
    global my_list
    my_list = [[1,2],[3,4]]
    list_comp_output = list_comp
    return list_comp_output


my_def([])
print my_def([x[1] for x in my_list])
[2,4]

此外，我在这里看不到任何合乎逻辑的东西 :) 如果您想使用全局变量，只需在模块范围内定义 if（在函数之外并将其传递给您的函数。）

def my_def(list_comp):
    # do stuff with list_comp
    return list_comp_output


my_list= # a costume list 
li=[x[1] for x in my_list]
print my_def(li)

或者更优雅地在你的函数中做列表理解：

def my_def(list_comp):
    return [x[1] for x in list_comp]

---

1. Mark Lutz 的《学习 Python》

Answer 2

那是因为 my_list 没有定义；）

首先你必须意识到函数定义中的主体（但是如果有默认参数，这些将立即被评估）直到函数被实际调用才被执行，但它的参数在函数之前被评估函数被调用。

其中的前两个语句首先表示符号my_list 是该函数中全局范围的符号。第二个表示为该符号（在全局范围内）分配值[[1,2],[3,4]]。但这确实在调用函数之前不会发生（所以在函数定义之后它仍然没有定义）。

这样做的结果是您尝试在定义或分配给（在函数体中）之前评估 my_list（在调用函数时）。

您可以尝试通过首先调用my_def([]) 来实现它，这将定义my_list，但正确的方法可能是将my_list 的定义放在函数定义之外：

my_list = [[1,2],[3,4]]

def my_def(list_comp):
   list_comp_output = list_comp
   return list_comp_output

print my_def([x[1] for x in my_list])

这给出了答案：

[2, 4]