การดึงแนวภาพยนตร์หลัก
คุณเพิ่งเข้าร่วมทีมข้อมูลของสตาร์ทอัพสตรีมมิงภาพยนตร์แห่งหนึ่ง ทีมนี้จัดเก็บข้อมูล metadata ของภาพยนตร์ไว้ในไฟล์ Parquet โดยคอลัมน์ genres ของแต่ละเรื่องเป็น list ของสตริง เนื่องจากภาพยนตร์ส่วนใหญ่มีหลายแนว งานของคุณคือดึงแนวภาพยนตร์แรกสุดออกมาจากแต่ละแถว แล้วสร้างเป็นคอลัมน์ระดับบนสุดชื่อ primary_genre
โหลด polars ไว้เป็น pl และโหลด DataFrame movies จากไฟล์ Parquet ไว้แล้ว
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
การปรับขนาดและเพิ่มประสิทธิภาพ Data Pipeline ด้วย Polars
คำแนะนำการฝึกหัด
- ดึงองค์ประกอบแรกจาก list ในคอลัมน์
genresของแต่ละแถว โดยใช้ list expression - ตั้งชื่อคอลัมน์ใหม่เป็น
primary_genreด้วย alias
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
result = movies.select(
"movie_title",
"genres",
# Get the first genre
pl.col("genres").list.____(0).alias("____"),
).head(8)
print(result)