요약

polars: select() 메서드와 정규표현식 활용

pandas: 열 list에 접근하여 Boolean 값 반환 후 df.loc으로 인덱싱

1. polars

# 조건 1개
music_df = df.select(pl.col('^music_.*$'))
movie_df = df.select(pl.col('^.*movie$'))
 
# 조건 2개 이상
fruits_df = df.select(pl.col('^apple.*$|^banana.*$'))
nutrient_df = df.select(pl.col('^.*brix$|^.*size$'))

2. pandas

조건 1개

start_cols = df.columns.str.startswith('music_')
end_cols   = df.columns.str.endswith('_movie')
 
music_df = df.loc[:, start_cols]
movie_df = df.loc[:, end_cols]

조건 2개 이상

조건 기호

^: ~로 시작하는 조건

$: ~로 끝나는 조건

start_cols = df.columns.str.contains('^apple_|^banana_', regex=True)
end_cols   = df.columns.str.contains('_brix$|_size$', regex=True)
 
fruits_df   = df.loc[:, start_cols]
nutrient_df = df.loc[:, end_cols]

참고사이트