Select columns by multiple partial string match from a pandas DataFrame

Question

I have the df with following columns:

df.columns

['total_rech_num_6', 'total_rech_num_7', 'total_rech_num_8','max_rech_amt_6', 'max_rech_amt_7', 'max_rech_amt_8','max_amt_6', 'max_amt_7', 'max_amt_8'].

I would like to select only those columns which has 'rech' AND '6' in the column name.

I tried this to select columns with'rech':

recharge_cols = [col for col in df.columns if 'rech' in col]

But i dont kow how to add the other string condition to get the 'rech' AND '6' columns. Need help!!

MNK · Accepted Answer · 2019-05-11 07:10:54Z

4

You can use this 1 line expression:

   recharge_cols = [i for i in list(df) if 'rech' in i and '6' in i]

answered May 11, 2019 at 7:10

MNK

6647 silver badges20 bronze badges

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

Comments

jezrael · Accepted Answer · 2019-05-11 07:08:10Z

1

Use and with another in statement:

recharge_cols = [col for col in df.columns if 'rech' in col and '6' in col]
print (recharge_cols)
['total_rech_num_6', 'max_rech_amt_6']

Pandas solution:

recharge_cols = df.columns[df.columns.str.contains('rech') & 
                           df.columns.str.contains('6')].tolist()

print (recharge_cols)
['total_rech_num_6', 'max_rech_amt_6']

answered May 11, 2019 at 7:08

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Comments

Klemen Koleša · Accepted Answer · 2019-05-11 07:08:42Z

0

Try this:

recharge_cols = [col for col in df.columns if 'rech' in col and '6' in col]

answered May 11, 2019 at 7:08

Klemen Koleša

4563 silver badges6 bronze badges

Collectives™ on Stack Overflow

Select columns by multiple partial string match from a pandas DataFrame

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related