How to Slice Numpy Arrays ?

[Yoriz]

I have a numpy array that is 840x2048, but I only want columns 1 and 51:1076. Every time I try to slice out these columns I fail. I feel like my brain is failing right now, and I should be able to do this.

I've tried:

Code: Select all

data2 = data[:,[1,51:1076]]    # Fail
data2 = data[:][:,(1,51:1076)] # Fail
data2 = data[np.ix_( [:],[1] )] # Worth a try


[setrofim]

IIRC, you can't index into non-consecutive, non-evenly-spaced ranges. So you'll probably have to do something like

Code: Select all

import numpy as np

data = np.random.rand(840, 2048)
data2 = np.hstack((data[:, 1:2], data[:, 51:1076]))


[tnknepp]

Thanks. I was trying to avoid stacking them, but I guess sometimes you have to use what you have. Thanks again,