如何獲得列表中所有可能的結果等於或小於 2 但等於或大於 1？

Question

例如：查找所有長度為3的可能列表，並且所有元素都等於或小於2但等於或大於1。（可以使用numpy）

input: len = 3, m = 1, n=2
output:[[1,1,1],[1,1,2],[1,2,1],[2,1,1],[2,2,1],[2,1,2],[1,2,2],[2,2,2]]

（有點像np.random.randint，但是我們希望得到所有的結果）

Answer 1

可能不是最好的方法，但你可以這樣做：

from itertools import permutations

s = [list(t) for t in {p for p in permutations([1, 1, 1, 2, 2, 2], 3)}]

print(s)

Output：

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