Python numpy array manipulations

Question

I need to somehow make that array:

[[639 190]
 [ 44   1]
 [ 71   4]
 ...,
 [863 347]
 [870 362]
 [831 359]]

look like this:

[[[639 190]]
 [[ 44   1]]
 [[ 71   4]]
 ...,
 [[863 347]]
 [[870 362]]
 [[831 359]]]

How would I do that? I'm new to numpy and I need it for my scientific experiment.

Divakar · Accepted Answer · 2017-07-07 05:53:27Z

4

Add a new axis with None/np.newaxis -

a[:,None,:] # Or simply a[:,None]

Sample run -

In [222]: a = np.random.randint(0,9,(4,3))

In [223]: a
Out[223]: 
array([[1, 6, 6],
       [4, 4, 5],
       [7, 4, 4],
       [4, 1, 3]])

In [224]: a[:,None]
Out[224]: 
array([[[1, 6, 6]],

       [[4, 4, 5]],

       [[7, 4, 4]],

       [[4, 1, 3]]])

answered Jul 7, 2017 at 5:53

Divakar

222k19 gold badges273 silver badges374 bronze badges

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

Comments

AGN Gazer · Accepted Answer · 2017-07-07 06:01:29Z

3

In addition to newaxis/None mentioned by @Divakar,

np.expand_dims(input_array, axis=1)

answered Jul 7, 2017 at 6:01

AGN Gazer

8,4272 gold badges31 silver badges49 bronze badges

Collectives™ on Stack Overflow

Python numpy array manipulations

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