python pandas select columns combining list and other column name

Question

I have created a list with 2 column names of a data frame.

Now I want to use this list to select this 2 columns along with other columns.

I have for example a data frame with 5 columns: A, B, C, D, E

I create a list that references A and B:

lst = ['A','B']

Then when I try to print it adding C I get an error.

print(df[lst,'C'])

TypeError: '(['A', 'B'], 'C')' is an invalid key

Which is the correct way to select the columns in a list along with others?

thanks!

Sachin Kohli · Accepted Answer · 2022-10-04 16:42:15Z

1

Need to pass that as list too;

df[lst+['C']]

answered Oct 4, 2022 at 16:42

Sachin Kohli

1,9961 gold badge4 silver badges6 bronze badges

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

1 Answer 1