pandas filtering column names

Question

I have this type of dataframe;

A = ["axa","axb","axc","axd","bxa","bxb","bxc","bxd","cxa".......]

My question is I have this type of data but there are more than 350 columns and for example i need only 'c' including column names in new dataframe. How can i do that?

new dataframe columns should look like this;

B = A[["axc","bxc","cxa","cxb","cxc","cxd","dxc","exc","fxc".......]]

score 0 · Accepted Answer · answered Oct 18 '22 at 08:08

0

Use for filter columns names with c by DataFrame.filter:

df2 = df.filter(like='c')

Or use list comprehension for filter columns names:

df2 = df[[x for x in df.columns if 'c' in x]]

answered Oct 18 '22 at 08:08

jezrael

822,522
95
1,334
1,252

1

filter is working very fast and good – DrakkarLeon Oct 18 '22 at 08:27

score 0 · Answer 2 · answered Oct 18 '22 at 08:09

0

You can do it easily using list comprehension:

new_df = df[[col for col in df.columns if 'c' in col]]

answered Oct 18 '22 at 08:09

Romain Capron

1,565
1
18
23

pandas filtering column names

2 Answers2