Question

如果我有两个字符串的元组列表列表。我想将它展平为非嵌套的元组列表，我可以这样做：

>>> from itertools import chain
>>> lst_of_lst_of_lst_of_tuples = [ [[('ab', 'cd'), ('ef', 'gh')], [('ij', 'kl'), ('mn', 'op')]], [[('qr', 'st'), ('uv', 'w')], [('x', 'y'), ('z', 'foobar')]] ]
>>> lllt = lst_of_lst_of_lst_of_tuples
>>> list(chain(*list(chain(*lllt))))
[('ab', 'cd'), ('ef', 'gh'), ('ij', 'kl'), ('mn', 'op'), ('qr', 'st'), ('uv', 'w'), ('x', 'y'), ('z', 'foobar')]

但是有没有另一种方法可以使用嵌套的list(chain(*lst_of_lst))解压缩到非嵌套的元组列表？

Answer 1

在解压缩迭代器之前，您不需要调用list：

list(chain(*chain(*lllt)))

最好使用chain.from_iterable而不是解压缩来处理迭代器，而不是使用*将它们实现为元组：

flatten = chain.from_iterable
list(flatten(flatten(lllt)))

Answer 2

你可以保持解包，直到你碰到元组：

def unpack_until(data, type_):
    for entry in data:
        if isinstance(entry, type_):
            yield entry
        else:
            yield from unpack_until(entry, type_)

然后：

>>> list(unpack_until(lllt, tuple))
[('ab', 'cd'),
 ('ef', 'gh'),
 ('ij', 'kl'),
 ('mn', 'op'),
 ('qr', 'st'),
 ('uv', 'w'),
 ('x', 'y'),
 ('z', 'foobar')]

有没有办法避免这么多的列表（链（* list_of_list））？

2 个答案: