如何使所有的数字组合低于某个整数的numpy数组

Question

我有一个整数的numpy数组，我想创建一个数组数组，其中包含它下面的所有数字组合。例如，[2,3]将产生

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

因为这将是所有数组的组合，其数字小于或等于[2,3]

Answer 1

您可以使用itertools.product：

from itertools import product

a,b = 2,3

list(product(np.arange(a)+1,np.arange(b)+1))

输出：

[(1, 1), (1, 2), (1, 3), (2, 1), (2, 2), (2, 3)]