是否有一种 pythonic 方式来引用 for 循环内的迭代器变量名称？

Question

下面是一个代码片段：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

...

val = carry
for l in filter(None, [l1, l2]):
    val += l.val
    l = l.next

l1 和 l2 是两个链表。 如果列表为空，则将其设置为None以便循环仅处理尚未完全遍历的链表。 我想使用循环来防止代码重复。 问题在第 4 行，在循环迭代时，l1 或 l2 都不会更新为列表的下一个节点，而是 l。 有谁知道解决这个问题的任何好的模式？ 如果之前有人问过这个问题，但我不知道搜索答案的正确术语，我深表歉意。 也许定义一个单独的函数或匿名函数是最好的方法？

Answer 1

当您执行l = l.next ，您只是在更改名称l指向的位置。 如果您创建一个包含列表节点的链接列表类，您将能够改变链接列表对象，而无需参考您之前为其分配的名称。

我添加了一些用于演示目的的方法。

class ListNode:
    def __init__(self, x, next=None):
        self.val = x
        self.next = next

    @classmethod
    def from_values(cls, values):
        try:
            first_val = next(values)
            return cls(first_val, cls.from_values(values))
        except StopIteration:
            return None

    def __iter__(self):
        yield self.val
        if self.next is not None:
            yield from self.next

class LinkedList:
    def __init__(self, values):
        self.head = ListNode.from_values(iter(values))

    def __bool__(self):
        return bool(self.head)

    def __str__(self):
        if self.head is None:
            return "<>"
        return "<{}>".format(", ".join(str(i) for i in self.head))

l1 = LinkedList([])
l2 = LinkedList([1, 2, 3])
print(l1)
print(l2)
for l in filter(None, [l1, l2]):
    l.head= l.head.next
print(l1)
print(l2)

有输出：

<>
<1, 2, 3>
<>
<2, 3>

Answer 2

我认为您正试图以一种尴尬的方式概括这段代码。 我会给ListNode一个__iter__方法（或类似的方法），然后使用它：

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

    def __iter__(self):
        cur_node = self

        while cur_node:
            yield cur_node.val
            cur_node = cur_node.next

完成后，可以迭代列表节点，您只需将l1和l2传递给sum ：

l1_sum = sum(l1)
l2_sum = sum(l2)