주요 장르 추출하기
여러분은 스트리밍 영화 스타트업의 데이터 팀에 합류했습니다. 이 팀은 영화 메타데이터를 Parquet 파일로 저장하며, 대부분의 영화가 여러 장르에 걸쳐 있기 때문에 각 영화의 genres는 문자열 리스트로 구성되어 있습니다. 각 행에서 첫 번째 장르를 추출해 primary_genre라는 최상위 컬럼으로 깔끔하게 정리해 보세요.
polars는 pl로 로드되어 있으며, DataFrame movies는 Parquet 파일에서 미리 로드되어 있습니다.
이 연습은 강의의 일부입니다
Polars로 데이터 파이프라인 확장 및 최적화하기
연습 안내
- 리스트 표현식을 사용해 각 행의
genres리스트에서 첫 번째 요소를 추출하세요. - 새 컬럼의 이름을
primary_genre로 지정하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
result = movies.select(
"movie_title",
"genres",
# Get the first genre
pl.col("genres").list.____(0).alias("____"),
).head(8)
print(result)