Pass column index and range simultaneously in Pandas iloc

Question

Suppose I have a dataframe with 100 columns. I want to select only 0,1 and 51-100 columns for all rows. I am trying something like this as of now:

df.iloc[:,         


        

Answer 1

You can do this without numpy.r_:

new_df = df.iloc[:, [0, 1] + list(range(51,101))]

Answer 2

I think you need numpy.r_ for concanecate indices:

df.iloc[:,np.r_[0,1,51:102]]

Sample:

df = pd.DataFrame([[0] * 10], columns=list('abcdefghij'))
print (df)
   a  b  c  d  e  f  g  h  i  j
0  0  0  0  0  0  0  0  0  0  0

print (df.iloc[:,np.r_[0,1,5:8]])
   a  b  f  g  h
0  0  0  0  0  0