how to extract required dimension from a numpy array

Question

Given that I have a numpy array of three dimension [3,500,500], how can I extract one dimension as [1,500,500]?

import numpy as np
my_array = np.ones((3,500,500),dtype=int,order='C')
print (my_array)

which one ones you need?

alko
– alko

2013-12-13 22:22:02 +00:00
Commented Dec 13, 2013 at 22:22 — alko
– alko, Commented Dec 13, 2013 at 22:22

alko · Accepted Answer · 2013-12-13 22:23:20Z

5

Those ones?

>>> my_array[0,:,:]
array([[1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       ...,
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1]])
>>> my_array[0,:,:].shape
(500, 500)

answered Dec 13, 2013 at 22:23

alko

48.7k12 gold badges99 silver badges105 bronze badges

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

1 Comment

user2357112 Over a year ago

You can just use my_array[0], or my_array[:1] if you want to keep the first axis.

Sagi Mann · Accepted Answer · 2018-09-05 08:21:33Z

1

maybe you mean transposed:

my_array = np.random.rand(3,3)

[[0.03639883, 0.47393186, 0.25823457],
 [0.3732702 , 0.43692148, 0.13893397],
 [0.11873897, 0.71504391, 0.34635178]]

my_array.T[0]:
[0.03639883, 0.3732702, 0.11873897]
my_array.T[1]:
[0.47393186 , 0.43692148, 0.13893397]
my_array.T[2]:
[0.11873897, 0.71504391, 0.34635178]

answered Sep 5, 2018 at 8:21

Sagi Mann

3,7089 gold badges45 silver badges84 bronze badges

Collectives™ on Stack Overflow

how to extract required dimension from a numpy array

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related