Nhóm theo nhiều cột với window functions

Để đi sâu hơn, bạn muốn phân tách nhu cầu thuê xe theo cả thời điểm trong ngày và thời tiết. Những giờ nắng ấm có thể khác với những giờ mát mẻ, nhiều mây. Hãy tạo một nhóm thời tiết đơn giản, rồi tính tổng số lượt thuê cho từng cặp giờ–thời tiết.

polars đã được nạp là pl. DataFrame bikes có sẵn với các cột time, rentals, temp, và hour.

Bài tập này là một phần của khóa học

Biến đổi dữ liệu với Polars

Xem khóa học

Hướng dẫn bài tập

Tạo cột weather: "warm" khi temp lớn hơn 20, ngược lại là "cool".
Thêm total_by_hour_weather là tổng rentals được nhóm theo hour và weather.

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.

# Add weather category based on temperature
bikes.with_columns(
    pl.when(pl.col("temp") > 20)
    .then(pl.lit("____"))
    .otherwise(pl.lit("____"))
    .alias("weather")
).with_columns(

    # Calculate total rentals by hour and weather
    pl.col("rentals").____().over("____", "____").alias("total_by_hour_weather")
)

Chỉnh sửa và Chạy Mã