Filling an array with arrays or vectors in python using numpy without a loop

Question

I'm trying to find a way to fill an array with rows of values. It's much easier to express my desired output with an example. Given the input of an N x M matrix, array1,

array1 = np.array([[2, 3, 4],
[4, 8, 3],
[7, 6, 3]])

I would like to output an array of arrays in which each row is an N x N consisting of the values from the respective row. The output would be

[[[2, 3, 4],
  [2, 3, 4],
  [2, 3, 4]],
 [[4, 8, 3],
  [4, 8, 3],
  [4, 8, 3]],
 [[7, 6, 3],
  [7, 6, 3],
  [7, 6, 3]]]

akuiper · Accepted Answer · 2017-03-06 04:33:03Z

1

You can reshape the array from 2d to 3d, then use numpy.repeat() along the desired axis:

np.repeat(array1[:, None, :], 3, axis=1)

#array([[[2, 3, 4],
#        [2, 3, 4],
#        [2, 3, 4]],

#       [[4, 8, 3],
#        [4, 8, 3],
#        [4, 8, 3]],

#       [[7, 6, 3],
#        [7, 6, 3],
#        [7, 6, 3]]])

Or equivalently you can use numpy.tile:

np.tile(array1[:, None, :], (1,3,1))

answered Mar 6, 2017 at 4:33

akuiper

216k33 gold badges362 silver badges379 bronze badges

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

Comments

Paul Panzer · Accepted Answer · 2017-03-06 05:11:41Z

1

Another solution which is sometimes useful is the following

out = np.empty((3,3,3), dtype=array1.dtype)
out[...] = array1[:, None, :]

answered Mar 6, 2017 at 5:11

Paul Panzer

53.3k3 gold badges59 silver badges103 bronze badges

Collectives™ on Stack Overflow

Filling an array with arrays or vectors in python using numpy without a loop

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related