I have the following dataframe that contains columns like: I would like the column names that contain _main sent to a list. such as: How best could I do that? Thanks very much! Answer try via filter() method: OR Via boolean masking:

try via filter() method:collist=df.filter(like='_main').columns.tolist()ORVia boolean masking:collist=df.columns[df.columns.str.endswith('_main')].tolist()

Send column names that contain a certain string to a list in pandas

I have the following dataframe that contains columns like:

df

        a   b   c   d   a_main   b_main   c_main   d_main
row
row2
row3

JavaScript
​x
 
df
​
        a   b   c   d   a_main   b_main   c_main   d_main
row
row2
row3
​

I would like the column names that contain _main sent to a list. such as:

collist = ['a_main'  , 'b_main'  , 'c_main'  , 'd_main']

JavaScript
 
collist = ['a_main'  , 'b_main'  , 'c_main'  , 'd_main']
​

How best could I do that? Thanks very much!

Answer

try via filter() method:

collist=df.filter(like='_main').columns.tolist()

JavaScript
 
collist=df.filter(like='_main').columns.tolist()
​

Via boolean masking:

collist=df.columns[df.columns.str.endswith('_main')].tolist()

JavaScript
 
collist=df.columns[df.columns.str.endswith('_main')].tolist()
​

Advertisement

Answer