"Se um trabalhador quiser fazer bem o seu trabalho, ele deve primeiro afiar suas ferramentas." - Confúcio, "Os Analectos de Confúcio. Lu Linggong"
Primeira página > Programação > Como agrupar dados por coluna e convertê-los em um dicionário de listas em Python?

Como agrupar dados por coluna e convertê-los em um dicionário de listas em Python?

Publicado em 2024-11-08
Navegar:497

How to Group Data by Column and Convert it to a Dictionary of Lists in Python?

GroupBy Results to Dictionary of Lists

Você tem uma planilha do Excel com três colunas, Coluna1, Coluna2 e Coluna3. Você deseja extrair os dados, agrupá-los pela Coluna1 e criar um dicionário de listas semelhante a este:

{0: [1],
1: [2, 3, 5],
2: [1, 2],
3: [4, 5],
4: [1],
5: [1, 2, 3]}

Para obter esse resultado, você pode usar o seguinte código:

excel = pandas.read_excel(r"e:\test_data.xlsx", sheetname='mySheet', parse_cols="A,C")
grouped_data = excel.groupby("Column1")["Column3"].apply(list).to_dict()

A função groupby agrupa os dados pela Coluna1 e retorna um objeto groupby. O método apply é então usado para aplicar a função de lista aos dados da Coluna3, criando uma lista de valores para cada grupo. Finalmente, o método to_dict é usado para converter os dados agrupados em um dicionário de listas.

O dicionário resultante terá o formato que você especificou, com as chaves sendo os valores na Coluna1 e os valores sendo listas da valores correspondentes na Coluna3.

Tutorial mais recente Mais>

Isenção de responsabilidade: Todos os recursos fornecidos são parcialmente provenientes da Internet. Se houver qualquer violação de seus direitos autorais ou outros direitos e interesses, explique os motivos detalhados e forneça prova de direitos autorais ou direitos e interesses e envie-a para o e-mail: [email protected]. Nós cuidaremos disso para você o mais rápido possível.

Copyright© 2022 湘ICP备2022001581号-3