Jak wydobyć nazwy kolumn z tabeli Pandas, których nazwy pasują do wzorca przynajmniej częściowo:

import pandas as pd

data = {'kot-1': [1,2,3], 'akuku kot': [4,5,6], 'kotara': [7,8,9], 'pies': [10,11,12]}
df = pd.DataFrame(data)

cat_cols = [col for col in df.columns if 'kot' in col] 

print(cat_cols)

Previous Post Next Post