How to make all number combinations of numbers below a certain numpy array of integers

Question

I have a certain numpy array of integers and I want to make an array of arrays that contains all combinations of numbers below it. As an example, [2,3] would yield

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

because that would be all the combinations of arrays with numbers less than or equal to [2,3].

Why there is `[3,3]` in your array? – Code Pope May 20 '20 at 15:45 — Code Pope, May 20 '20 at 15:45

score 1 · Answer 1 · answered May 20 '20 at 15:44

1

You can use itertools.product:

from itertools import product

a,b = 2,3

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

Output:

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

answered May 20 '20 at 15:44

Quang Hoang

146,074
10
56
74

For some reason she also has `[3,3]` in the list – Code Pope May 20 '20 at 15:46
Is that readily available on spyder? Thank you! – May 20 '20 at 15:48
Spyder is just an IDE, depending on your python. Should be available. – Quang Hoang May 20 '20 at 15:49
Great, thanks. You were very helpful! – May 20 '20 at 15:49
Is there a way to make the argument of product an array so I can make it any size? Sorry for the inconvenience. – May 20 '20 at 15:53

How to make all number combinations of numbers below a certain numpy array of integers

1 Answers1