Selecting all columns in DataFrame.set_index except one

Question

As the title says, I'm trying to select all columns except one in DataFrame.set_index.

I tried the following way:

df = df.set_index(list(df.columns != 'cus_name'))

The cus_name Series is the one I want to exclude. The above code raise a KeyError: True.

The list(df.columns != 'cus_name') is a list of boolean values [True, True, False, True, True, True, True, True, True, True, True, True] and what I need is a list of columns names except the cus_name.

I know I could explicitly input the complete list of columns I want in the set_index method but I was wandering if there is a more efficient way to do this.

anky · Accepted Answer · 2019-07-08 12:00:14Z

6

You can use pd.Index.difference() here with sort=False if order is important:

df=df.set_index(df.columns.difference(['cus_name'],sort=False).tolist())

answered Jul 8, 2019 at 12:00

anky

75.3k11 gold badges46 silver badges76 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Josef Korbel · Accepted Answer · 2019-07-08 11:57:34Z

4

Try list comprehension

df = df.set_index([c for c in df.columns if c != 'cus_name'])

answered Jul 8, 2019 at 11:57

Josef Korbel

1,2081 gold badge10 silver badges32 bronze badges

Comments

DeepSpace · Accepted Answer · 2019-07-08 11:59:39Z

3

A cheeky way to achieve this with set operations:

df.set_index(list(set(df.columns) - {'cus_name'}))

answered Jul 8, 2019 at 11:59

DeepSpace

82.1k12 gold badges119 silver badges166 bronze badges

Comments

Mark Wang · Accepted Answer · 2019-07-08 12:02:13Z

1

Alternatively,

df = df.set_index(df.columns.drop('cus_name').tolist()))

answered Jul 8, 2019 at 12:02

Mark Wang

2,7579 silver badges18 bronze badges

Collectives™ on Stack Overflow

Selecting all columns in DataFrame.set_index except one

4 Answers 4

Comments

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related