Python 我应该使用哪个 itertools 模块函数？

Question

如果我想实现这一点，我应该使用哪个 itertools 函数（或任何函数）：

输入：

arr = [0, 1, 2]

输出：

[
[0, 0, 0]
[0, 0, 1]
[0, 0, 2]
[0, 1, 1]
[0, 1, 2]
[0, 2, 2]
[1, 1, 1]
[1, 1, 2]
[1, 2, 2]
[2, 2, 2]
]

条件：

• r/3 的长度
• 没有例如的组合1,0,0 和 0,1,0
• 是的重复值，例如1,1,1 或 0,1,1

Answer 1

print([a for a in itertools.combinations_with_replacement(arr, 3)])