选择性展平Python列表

Question

假设我有一个包含（以及其他）不同类型的子列表的列表：

[1, 2, [3, 4], {5, 6}]

我想以一种选择性的方式扁平化，这取决于它的元素类型（即我只想展平sets ，而其余​​部分保持不平整）：

[1, 2, [3, 4], 5, 6]

我目前的解决方案是一个功能，但仅仅是出于我的求知欲，我想知道是否可以用一个列表理解来做到这一点？

Answer 1

列表推导不是为扁平化而设计的（因为它们没有办法组合对应于多个输入项的值）。

虽然您可以使用嵌套列表推导来解决这个问题，但这需要您的顶级列表中的每个元素都是可迭代的。

老实说，只需使用一个功能。 这是最干净的方式。

Answer 2

琥珀可能是正确的，因为这样的功能更适合。 另一方面，总有一点变化的空间。 我假设嵌套永远不会超过一个级别 - 如果它的深度超过一个级别，那么你肯定更喜欢这个功能。 但如果没有，这是一种潜在可行的方法。

>>> from itertools import chain
>>> from collections import Set
>>> list(chain.from_iterable(x if isinstance(x, Set) else (x,) for x in l))
[1, 2, [3, 4], 5, 6]

非itertools的方法将涉及嵌套列表推导。 最好将其分为两行：

>>> packaged = (x if isinstance(x, collections.Set) else (x,) for x in l)
>>> [x for y in packaged for x in y]
[1, 2, [3, 4], 5, 6]

我对这些中的任何一个是否比直接函数更快或更慢都没有强烈的直觉。 这些会产生很多单例元组 - 这有点浪费 - 但它们也是以LC速度发生的，这通常都很好。

Answer 3

您可以使用funcy库中的flatten函数：

from funcy import flatten, isa

flat_list = flatten(your_list, follow=isa(set))

您还可以查看其实现 。