Selecting first n columns and last n columns with pandas

Question

I am trying to select the first 2 columns and the last 2 column from a data frame by index with pandas and save it on the same dataframe.

is there a way to do that in one step?

Answer 1

You can use the iloc function to get the columns, and then pass in the indexes.

df.iloc[:,[0,1,-1,-2]]