Extraer el género principal

Te has incorporado al equipo de datos de una startup de streaming de películas. Guardan los metadatos de los filmes en un archivo Parquet donde el genres de cada película es una lista de cadenas, ya que la mayoría abarca varios géneros. Extrae el primer género listado de cada fila a una columna de nivel superior, limpia, llamada primary_genre.

polars está cargado como pl, y el DataFrame movies está precargado desde el archivo Parquet.

Este ejercicio forma parte del curso

Escala y optimiza canalizaciones de datos con Polars

Ver curso

Instrucciones del ejercicio

Extrae el primer elemento de la lista genres de cada fila usando una expresión de lista.
Pon como alias la nueva columna primary_genre.

ejercicio interactivo práctico

Prueba este ejercicio completando este código de ejemplo.

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

Editar y ejecutar código