使用itertools在Python中的Powersets

Question

我正在尝试在Python 3中创建一个powerset.我找到了对该itertools 模块的引用,并且我使用了该页面上提供的powerset代码.问题是:代码返回itertools.chain对象的引用,而我想访问powerset中的元素.我的问题:如何做到这一点？

非常感谢您的见解.

Answer 1

itertools函数返回迭代器,按需懒惰地生成结果的对象.

您可以使用循环遍历对象,也可以for通过调用将结果转换为列表list():

from itertools import chain, combinations

def powerset(iterable):
    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
    s = list(iterable)
    return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))

for result in powerset([1, 2, 3]):
    print(result)

results = list(powerset([1, 2, 3]))
print(results)

您还可以将对象存储在变量中,并使用该next()函数逐个从迭代器获取结果.