是否有一個 Python 函數可以找到列表的所有 k 長度排序？

Question

我不相信以前有人問過這個確切的問題。 我最近遇到了一個問題，我必須找到這樣一個集合。 一個例子可能會有所幫助：-

給出一些列表：

list1 = ['a', 'b']

是否有返回以下集合的函數？

output = {('a', 'b'), ('a', 'a'), ('b', 'b'), ('b', 'a')}

我已經能夠產生使用所期望的輸出itertools combinations_with_replacement和permutations功能，如下所示：

from itertools import combinations_with_replacement, permutations
set1 = set(combinations_with_replacement(['a', 'b'], 2))
set2 = set(permutations(['a', 'b'], 2))

>>> set1
{('a', 'b'), ('a', 'a'), ('b', 'b')}
>>> set2
{('b', 'a'), ('a', 'b')}

set1.update(set2)

>>> set1
{('a', 'b'), ('a', 'a'), ('b', 'b'), ('b', 'a')}

有這樣一套的名字嗎？ 有沒有我可以使用的替代方法？

Answer 1

你想要itertools.product ：

>>> import itertools
>>> set(itertools.product(list1, repeat=2))
{('a', 'b'), ('b', 'a'), ('b', 'b'), ('a', 'a')}

帶有repeat參數的itertools.product本質上是“ permutations_with_replacement ”，這似乎是您想要的。

Answer 2

Itertools.product() 做你想要的：

mylist = ['a', 'b']
list(itertools.product(mylist, repeat=2))

Out[8]: [('a', 'a'), ('a', 'b'), ('b', 'a'), ('b', 'b')]