如何通过过滤旧元素将多个元素附加到新列表

Question

我有一个像[ ((xs,xe),idx),...]

这样的列表

我想将xs和xe收集到一个集合中，就像这样

s={ xs,xe for ((xs,xe),idx)

我知道我可以使用它两次并分别添加xs，xe。

有更好的方法吗？

Answer 1

您可以使用itertools.chain.from_iterable

>>> import itertools
>>> lst = [((1,2), 'a'), ((3,4), 'b')]
>>> s = set(itertools.chain.from_iterable(pair for pair, _ in lst))
>>> s
set([1, 2, 3, 4])