Extraindo o gênero principal
Você entrou para a equipe de dados de uma startup de streaming de filmes. Eles armazenam metadados dos filmes em um arquivo Parquet em que o genres de cada filme é uma lista de strings, já que a maioria dos títulos se encaixa em vários gêneros. Extraia o primeiro gênero listado de cada linha para uma coluna de nível superior, limpa, chamada primary_genre.
polars está carregado como pl, e o DataFrame movies já foi carregado a partir do arquivo Parquet.
Este exercicio faz parte do curso
Dimensionamento e Otimização de Pipelines de Dados com Polars
Instruções do exercicio
- Extraia o primeiro elemento da lista
genresde cada linha usando uma expressão de lista. - Dê o alias da nova coluna como
primary_genre.
exercicio interativo prático
Tente este exercicio completando este código de exemplo.
result = movies.select(
"movie_title",
"genres",
# Get the first genre
pl.col("genres").list.____(0).alias("____"),
).head(8)
print(result)