How to merge two arrays of same size into an array of tuples with numpy?

Asked Oct 19 '19 at 16:31

Active Oct 19 '19 at 16:31

Viewed 360 times

I have a project in python where I want to get a 2D matrix filled with coordinates:

First, I use the meshgrid function to get my coordinates in 2 arrays:

X, Y = np.meshgrid(np.arange(5), np.arange(5))

I get:

X =   0 1 2 3 4        Y =   0 0 0 0 0
      0 1 2 3 4              1 1 1 1 1
      0 1 2 3 4              2 2 2 2 2
      0 1 2 3 4              3 3 3 3 3
      0 1 2 3 4              4 4 4 4 4

Now, I'd like to merge the two arrays into a single array of tuples like this:

Coordinates = (0,0) (1,0) (2,0) (3,0) (4,0) 
              (0,1) (1,1) (2,1) (3,1) (4,1) 
              (0,2) (1,2) (2,2) (3,2) (4,2) 
              (0,3) (1,3) (2,3) (3,3) (4,3) 
              (0,4) (1,4) (2,4) (3,4) (4,4)

And I want to create a really big matrix, like 10 000 x 10 000 , so it has to be really fast (as fast as possible in fact) ...

asked Oct 19 '19 at 16:31

rambi

1,013
1
7
25

Nice ! Thank you – rambi Oct 19 '19 at 16:42
Would you be happy with a 3d array, with the last dimension of 2? – hpaulj Oct 19 '19 at 16:46
```np.dstack((X,Y))``` would work in this case. – Grzegorz Skibinski Oct 19 '19 at 16:48
1

`X + 1j*Y` makes a 2d array of complex numbers. – hpaulj Oct 20 '19 at 18:16
@hpaulj Perfect ! – rambi Oct 20 '19 at 18:29
Thank you for your answers ! Finally, I found a simpler way by using complex numbers: `X, Y = np.meshgrid(np.arange(5), np.arange(5)) matrix = X + Y * 1j` What I get: `[[0.+0.j 1.+0.j 2.+0.j 3.+0.j 4.+0.j] [0.+1.j 1.+1.j 2.+1.j 3.+1.j 4.+1.j] [0.+2.j 1.+2.j 2.+2.j 3.+2.j 4.+2.j] [0.+3.j 1.+3.j 2.+3.j 3.+3.j 4.+3.j] [0.+4.j 1.+4.j 2.+4.j 3.+4.j 4.+4.j]] ` – rambi Oct 21 '19 at 15:05

How to merge two arrays of same size into an array of tuples with numpy?

0 Answers0