24 lines
921 B
Python
24 lines
921 B
Python
from itertools import permutations, combinations
|
||
|
||
if __name__ == '__main__':
|
||
# 有时候我们想对可迭代对象中的元素进行排列组合，这个时候我们就不得不靠for循环写出很不python的代码
|
||
# itertools库帮助我们解决了这个困扰
|
||
items= ['a', 'b', 'c', 'd', 'e']
|
||
|
||
# permutations会列出所有的排列，如果指定排列长度，那就会输出该长度下所有的排列
|
||
for p in permutations(items):
|
||
print(p)
|
||
|
||
for p in permutations(items, 3):
|
||
print(p)
|
||
|
||
# 如果想要得到特定长度的组合，那就可以使用combination函数
|
||
for c in combinations(items, 3):
|
||
print(c)
|
||
|
||
# 上面这种组合是不放回的组合，如果想产生放回元素的组合，可以使用combinations_with_replacement函数
|
||
from itertools import combinations_with_replacement as cwr
|
||
for c in cwr(items, 3):
|
||
print(c)
|
||
|