Return list of columns with string match in multiple DataFrames

Question

I have some DataFrames:

d = {'colA': [1, 2], 'colB': [3, 4]}

df = pd.DataFrame(data=d)
df2 = pd.DataFrame(data=d)
df3 = pd.DataFrame(data=d)

I want to return a list of columns containing the string 'A', e.g. for one DataFrame:

[column for column in df.columns if 'A' in column]

How can I do this for multiple DataFrames (e.g., df, df2, df3)?

The desired output in this example would be ['colA', 'colA', 'colA']

rhug123 · Accepted Answer · 2021-06-10 23:52:46Z

1

Here is a way:

l = [','.join(i.columns[i.columns.str.contains('A')]) for i in [df,df2,df3]]

answered Jun 10, 2021 at 23:52

rhug123

8,8801 gold badge14 silver badges27 bronze badges

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

Comments

Inputvector · Accepted Answer · 2021-06-10 23:48:33Z

1

You can create a function for that:

def find_char(list_of_df , char):
    result = []
    for df in list_of_df:
        for c in df.columns:
            if char in c:
                result.append(c)
    return result

Usage:

res = find_char([df,df2,df3] , 'A')
print(res)
['colA', 'colA', 'colA']

answered Jun 10, 2021 at 23:48

Inputvector

1,09310 silver badges22 bronze badges

Collectives™ on Stack Overflow

Return list of columns with string match in multiple DataFrames

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related