Numpy meshgrid 2d-array

Question

I want to create a meshgrid and I only have one 2-dimensional array.

I want to create this:

a = np.array([1,2])
b = np.array([3,4])
mesh = np.array(np.meshgrid(a, b))
## mesh = 
## array([[[1, 2],
##    [1, 2]],
##
##   [[3, 3],
##    [4, 4]]])

but the input is one array containing a and b, like this:

ab = np.array([[1,2],[3,4]])
mesh = np.array(np.meshgrid(ab))
## mesh = 
## array([[[1, 2],
##    [1, 2]],
##
##   [[3, 3],
##    [4, 4]]])

Is their a way to archive that even with 3 or 4 arrays packed in one big one like this:

abcd = np.array([[1,2],[3,4],[5,6],[7,8]])

Quang Hoang · Accepted Answer · 2020-07-19 13:15:01Z

3

Yes, we can unpack abcd and pass to messgrid:

mesh = np.array(np.meshgrid(*abcd))

answered Jul 19, 2020 at 13:15

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Numpy meshgrid 2d-array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related