การ Pivot DataFrame

ต้องการแปลงข้อมูล Spotify จากรูปแบบ long format เป็น wide format เพื่อเปรียบเทียบจำนวนสตรีมของแต่ละศิลปินในแต่ละปี

DataFrame spotify_df พร้อมใช้งานแล้ว และได้ถูกปรับให้แสดงจำนวนสตรีมของแต่ละศิลปินในแต่ละปีไว้ใน artist_year_df

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

Introduction to Polars

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

artist_year_df = spotify_df.group_by("artist", "year").agg(pl.col("streams").sum())

pivoted_df = (
    artist_year_df
    # Pivot with artists as rows, years as columns and streams as values
    .pivot(
        on=____,
        index=____,
        values=____
    )
)

print(pivoted_df)

แก้ไขและรันโค้ด