如何在列表理解中执行作业？

Question

因此，在 Python 3.6 中为 tress 创建测试用例时，我想知道是否有一种很好的方法来执行列表理解中的赋值等操作。 我应该使用 lambda 吗？

我想插入

Prev_Node.next = Node(i)
Prev_Node = Node(i)

进入

iList = [1,2,3,4,10]
Prev_Node = Node(iList[0])
l1 = [Node(i) <Any assignment action?> for i in iList[1:])]

提前感谢您的时间，

Answer 1

使用迭代器遍历您的列表，不要摆弄.Next等来构建您自己的列表。

如果您正在寻找列表头和尾，那么您可以执行以下操作：

int_list = [1, 2, 3, 4, 10]
node_list = [Node(i) for i in int_list]
head = node_list[0]  # Get the head Node and leave it in the list.
head, *tail = node_list  # Get the head Node and remove it from the list.

并像这样迭代

for node in node_list:
    print(node)

查看有关如何进一步操作现有列表的各种列表功能。

Answer 2

你不能在列表理解中做作业。 尝试这样做会引发`SyntaxError，例如：

[x = 0 for _ in range(1)]
# SyntaxError: invalid syntax

我想尝试实现这一点的一个可怕的技巧可能是定义一个函数，您可以在列表理解中调用该函数来进行赋值（使用全局变量）。 但这比使用普通的 for 循环而不是列表理解要糟糕得多。 基本上，任何时候使用列表推导式无法立即工作，您可能最好使用 for 循环。 你会更容易编程，也更容易让其他人理解。

# hacky; don't do this
def set_x(val):
    global x
    x = val

[set_x(i) for i in range(3)]
x # 2