Trích xuất thể loại chính
Bạn vừa gia nhập đội dữ liệu tại một startup streaming phim. Họ lưu siêu dữ liệu phim trong một tệp Parquet, trong đó genres của mỗi phim là một danh sách chuỗi vì hầu hết phim thuộc nhiều thể loại. Hãy lấy thể loại đầu tiên trong danh sách của mỗi hàng và đưa vào một cột cấp cao primary_genre gọn gàng.
polars đã được nạp dưới tên pl, và DataFrame movies đã được tải sẵn từ tệp Parquet.
Bài tập này là một phần của khóa học
Mở rộng và tối ưu hóa pipeline dữ liệu với Polars
Hướng dẫn bài tập
- Trích xuất phần tử đầu tiên từ danh sách
genrescủa mỗi hàng bằng biểu thức danh sách. - Đặt bí danh cho cột mới là
primary_genre.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
result = movies.select(
"movie_title",
"genres",
# Get the first genre
pl.col("genres").list.____(0).alias("____"),
).head(8)
print(result)