Multiple selection of dataframe using multiple column slices

Question

I have a dataframe that has 10 columns, I am trying select columns 1 to 3 (first to third) and columns 5 to 8 (fifth to eighth). Is it possible to select using the iloc selector so that it can take multiple slices. The sample below can slice first to third, I would like to add the columns 5 to 8 as well

df = pc_df.iloc[:,0:4]

score 2 · Accepted Answer · answered Jan 14 '22 at 13:31

2

You can use numpy:

#import  numpy as np
df.iloc[:,np.r_[0:4, 5:9]]

np.r_ will concatenate the indexes for you.

answered Jan 14 '22 at 13:31

TitoOrt

1,265
1
11
13

Been trying to figure this out for a while just couldn't get my head around it. This np.r_ solved it. – mapperx Jan 14 '22 at 13:40

score 2 · Answer 2 · answered Jan 14 '22 at 13:44

2

Didn't know about np.r_, so also thanks to Tito from me!

If you don't want to use extra lib, you could also use pd.concat

pd.concat([df.iloc[:,:4], df.iloc[:, 5:9]],1)

answered Jan 14 '22 at 13:44

Rabinzel

7,757
3
10
30

Worked as well. I had tried something similar but missed out the concat, the enclosing square bracket and the 1 at the end. – mapperx Jan 14 '22 at 14:00
The np.r_ is new to me too, will add it to my reading list. – mapperx Jan 14 '22 at 14:01

Multiple selection of dataframe using multiple column slices

2 Answers2