building a 3d Numpy array from a 2d numpy array

Question

So I've got a numpy array like this:

a = np.array([[1, 2, 3],
              [2, 3, 4],
              [4, 5, 6]])

and I want to convert it into an array like this:

[[[1, 1, 1], [2, 2, 2], [3, 3, 3]],
 [[2, 2, 2], [3, 3, 3], [4, 4, 4]],
 [[4, 4, 4], [5, 5, 5], [6, 6, 6]]]

How would you do it?

mozway · Accepted Answer · 2022-06-29 20:12:00Z

3

Use numpy.repeat on the array with one extra dimension:

np.repeat(a[...,None], 3, axis=2)

Or numpy.tile:

np.tile(a[...,None], (1,1,3))

Output:

array([[[1, 1, 1],
        [2, 2, 2],
        [3, 3, 3]],

       [[2, 2, 2],
        [3, 3, 3],
        [4, 4, 4]],

       [[4, 4, 4],
        [5, 5, 5],
        [6, 6, 6]]])

edited Jun 29, 2022 at 20:12

answered Jun 29, 2022 at 20:06

mozway

267k13 gold badges56 silver badges106 bronze badges

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

Collectives™ on Stack Overflow

building a 3d Numpy array from a 2d numpy 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