Select specific columns in NumPy array using colon notation

Question

I have a 40000 by 60 Numpy array, and I want to sth like this:

mat[:,[0:13,19:23,23:31,39:59]]

Obviously, it does not work. Is there a smarter way to do this than concatenation?

Divakar · Accepted Answer · 2017-06-05 19:02:08Z

5

mat[:,np.r_[0:13,19:23,23:31,39:59]]

Sample run -

In [48]: mat = np.random.rand(100,1000)

In [50]: mat[:,np.r_[0:13,19:23,23:31,39:59]].shape
Out[50]: (100, 45)

In [51]: 13+4+8+20
Out[51]: 45

answered Jun 5, 2017 at 19:02

Divakar

222k19 gold badges273 silver badges374 bronze badges

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

1 Answer 1