python - 将列表组合成一个可以重复使用的可迭代对象

Question

我有一个问题，我有一些列表(2 个或更多)需要组合成一个可迭代对象。问题是，构成可迭代对象的列表在不断更新，我希望我的可迭代对象无需任何额外工作即可获得这些更改。

仅仅组合列表是行不通的，因为将一个列表添加到一个列表会生成一个新列表。

list_a = ['foo', 'bar']
list_b = ['abc']
list_c = list_a + list_b
list_a.append('ttt')
print(list_c)
# Result: ['foo', 'bar', 'abc']

我可以将列表添加到一个列表中以创建一个列表列表，这可行，但解压缩它会使我的生产代码的逻辑变得过于复杂而无法实现。

list_a = ['foo', 'bar']
list_b = ['abc']
list_c = [list_a, list_b]
list_a.append('ttt')
print(list_c)
# Result: [['foo', 'bar', 'ttt'], ['abc']]

我喜欢 itertools.chain 的想法，因为它让我参与其中，但在我失去对原始列表的引用之前，我只能遍历列表一次。

import itertools

list_a = ['foo', 'bar']
list_b = ['abc']
iter_c = itertools.chain(list_a, list_b)
list_a.append('ttt')

for item in iter_c:
    print(item)  # Works fine here

for item in iter_c:
    print(item)  # The iterable was exhausted - this doesn't work anymore

您可以将链转换为列表，但一旦您这样做，对 list_a 的修改将不会继续。

import itertools

list_a = ['foo', 'bar']
list_b = ['abc']
iter_c = itertools.chain(list_a, list_b)
list_a.append('ttt')
list_c = list(iter_c)

for item in list_c:
    print(item)  # Works fine here

for item in list_c:
    print(item)  # Now works fine here, too

list_a.append('zzz')  # This won't get added to our chain
print(list_c)
# Result: ['foo', 'bar', 'ttt', 'abc']

我建立了一个 hacky 类来做我想做的事，但我对它非常不满意。

import collections
import uuid


class IterGroup(object):
    def __init__(self):
        super(IterGroup, self).__init__()
        self._data = collections.OrderedDict()

    def append(self, item):
        # The key doesn't matter as long as it's unique. The key is ignored.
        self._data[str(uuid.uuid4())] = item

    def __iter__(self):
        for items in self._data.values():
            for item in items:
                yield item


list_a = ['foo', 'bar']
list_b = ['abc']
list_c = IterGroup()  # Not really a list but just go with it
list_c.append(list_a)
list_c.append(list_b)

list_a.append('ttt')

for item in list_c:
    print(item)

list_a.append('zzz')

for item in list_c:
    print(item)
# Prints ['foo', 'bar', 'ttt', 'zzz', 'abc']

所以我对所需解决方案的标准是

1. 必须能够向其添加可迭代对象
2. 在添加其中一个可迭代对象后对其进行修改将自动反射(reflect)在组解决方案中
3. 组合的可迭代对象可以迭代任意多次
4. 必须维持秩序
5. 理想情况下，我也可以按索引获取项目，尽管这不是必需的
6. 我宁愿解决方案不是自定义类(但如果必须如此，我也不会挑剔)
7. 必须使用 Python 2

有没有人对这个问题有一个干净的解决方案？

Answer 1

我不认为有一种方法可以针对这种行为使用自定义类，但我不明白为什么该类应该扩展 OrderedDict；只需实现 __iter__ 和 __getitem__ 方法。你可以尝试这样的事情。

class Multilistview:

    def __init__(self, *lists):
        self.lists = lists

    def __iter__(self):
        return itertools.chain.from_iterable(self.lists)

    def __getitem__(self, idx):
        if isinstance(idx, slice):
            return list(itertools.islice(self, idx.start, idx.stop, idx.step))
        else:
            for i, x in enumerate(self):
                if i == idx:
                    return x

或更简单，但每次您请求一个项目时，这都会具体化整个列表:

    def __getitem__(self, idx):
        return list(self)[idx]

(您还可以通过检查每个项目的长度来确定要使用的列表以及该列表中的“更正”索引，从而使 __getitem__ 更加高效。)

例子

list_a = ['foo', 'bar']
list_b = ['abc']
list_c = Multilistview(list_a, list_b)
for x in list_c:
    print(x)
# foo
# bar
# abc
list_a.append('blub')
list_b[:] = [1,2,3]
print(list(list_c))
# ['foo', 'bar', 'blub', 1, 2, 3]
print(list_c[4])
# 2
print(list_c[2:5])
# ['blub', 1, 2]