Select specific columns in a NumPy array using colon

Question

I have a 40,000 by 60 Numpy array and I want it like this:

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

Obviously this won't work. Is there a smarter way to do this than concatenation?

+3

Gordon cai 05 june 17 at 19:01

1 answer

Divakar · Accepted Answer · 2017-06-05T19:02:08+0000

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

Example 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