如何获得列表中所有可能的结果等于或小于 2 但等于或大于 1？

Question

For example: To find all possible lists with length=3, and all elements are equal or smaller than 2 but equal or greater than 1. (you can use 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]]

(it's a little bit like np.random.randint, but we hope to get all results)

Answer 1

Probably not the best way but you could do this:

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]]