11

我有一个可变序列的迭代器,例如

foo = [1, 2, 3, 4, 5]
for bar in foo:

有没有办法通过使用迭代器中包含的引用来写入 foo 中的元素?天真的任务:

   bar = 42

当然不起作用。是否可以对迭代器中的序列元素使用“幕后”引用?

PS:使用索引的简单解决方案

for i in range(len(a)):
   a[i] = 42

不适用于我的情况,因为我无法公开容器名称。

4

3 回答 3

23

用于enumerate()在循环中为您生成索引:

for i, bar in enumerate(foo):
    foo[i] = bar + 42
于 2013-10-24T09:42:13.857 回答
9

据我了解,您的用例是这样的:

class Z:
    def __init__(self):
        self.a, self.b, self.c = 1, 2, 3

    def it(self):
        for x in self.a, self.b, self.c:
            yield x

z = Z()
for x in z.it():
    if x == 1:
       x = 42 # z.a should be 42? - doesn't work!

这在 python 中是不可能的——没有“指针”或“引用”数据类型。您可以通过产生一个 setter 函数而不是(或连同)该值来解决此问题:

class Z:
    def __init__(self):
        self.a, self.b, self.c = 1,2,3

    def it(self):
        for x in 'abc':
            yield getattr(self, x), lambda y: setattr(self, x, y)

z = Z()
for x, setter in z.it():
    if x == 1:
       setter(42) # works!
于 2013-10-24T10:11:43.567 回答
5

您可以使用列表理解

foo[:] = [expression for bar in foo]

或者,如果赋值对于表达式来说太复杂,您可以使用函数func

foo[:] = [func(bar) for bar in foo]

如果您只有一个迭代器,则无法将值重新分配给底层容器,因为甚至无法保证存在底层容器:

def generator():
    for i in range(10):
        yield i

for bar in generator():
    # Nothing you put here can change what's going on in generator
于 2013-10-24T09:44:45.880 回答