Mengambil genre utama

Anda bergabung dengan tim data di sebuah startup layanan streaming film. Mereka menyimpan metadata film dalam berkas Parquet, di mana genres setiap film berupa list string karena sebagian besar film mencakup beberapa genre. Ambil genre pertama yang tercantum dari setiap baris ke dalam kolom tingkat atas primary_genre yang rapi.

polars dimuat sebagai pl, dan DataFrame movies sudah dimuat dari berkas Parquet.

Latihan ini merupakan bagian dari kursus

Menskalakan dan Mengoptimalkan Pipeline Data dengan Polars

Lihat Kursus

Instruksi latihan

Ekstrak elemen pertama dari list genres di setiap baris menggunakan ekspresi list.
Beri alias kolom baru sebagai primary_genre.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

result = movies.select(
    "movie_title",
    "genres",
    # Get the first genre
    pl.col("genres").list.____(0).alias("____"),
).head(8)
print(result)

Edit dan Jalankan Kode