Pandas: converta rapidamente um dataframe em um dicionário

Você pode usar a seguinte sintaxe para converter um DataFrame do pandas em um dicionário:

 df. to_dict ()

Observe que to_dict() aceita os seguintes argumentos potenciais:

• dict: (padrão) Chaves são nomes de colunas. Os valores são dicionários de pares índice:dados.
• lista: chaves são nomes de colunas. Os valores são listas de dados de colunas.
• série: chaves são nomes de colunas. Os valores são séries de dados de coluna.
• divisão: as chaves são “colunas”, “dados” e “índice”.
• registros: chaves são nomes de colunas. Os valores são dados em células.
• índice: chaves são rótulos de índice. Os valores são dados em células.

Os exemplos a seguir mostram como usar essa sintaxe na prática com o seguinte DataFrame do pandas:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'B', 'B', 'C'],
                   ' points ': [5, 7, 9, 12, 9],
                   ' rebounds ': [11, 8, 6, 6, 5]})

#view DataFrame
df

	team points rebounds
0 to 5 11
1 to 7 8
2 B 9 6
3 B 12 6
4 C 9 5

Exemplo 1: Converter um DataFrame em um dicionário (“dict”)

O código a seguir mostra como converter um DataFrame do pandas em um dicionário usando o método padrão ‘ dict ‘:

 df. to_dict ()

{'team': {0: 'A', 1: 'A', 2: 'B', 3: 'B', 4: 'C'},
 'points': {0:5, 1:7, 2:9, 3:12, 4:9},
 'rebounds': {0:11, 1:8, 2:6, 3:6, 4:5}}

Exemplo 2: Converter um DataFrame em um Dicionário (“Lista”)

O código a seguir mostra como converter um DataFrame do pandas em um dicionário usando o método ‘ list ‘:

 df. to_dict (' list ')

{'team': ['A', 'A', 'B', 'B', 'C'],
 'points': [5, 7, 9, 12, 9],
 'rebounds': [11, 8, 6, 6, 5]}

Exemplo 3: Converter um DataFrame em um dicionário (“série”)

O código a seguir mostra como converter um DataFrame do pandas em um dicionário usando o método ‘ series ‘:

 df. to_dict (' series ')

{'team': 0 A
 1A
 2 B
 3 B
 4C
 Name: team, dtype: object,
 'points': 0 5
 1 7
 2 9
 3 12
 4 9
 Name: points, dtype: int64,
 'rebounds': 0 11
 1 8
 2 6
 3 6
 4 5
 Name: rebounds, dtype: int64}

Exemplo 4: Converter um DataFrame em um dicionário (“split”)

O código a seguir mostra como converter um DataFrame do pandas em um dicionário usando o método ‘ split ‘:

 df. to_dict (' split ')

{'index': [0, 1, 2, 3, 4],
 'columns': ['team', 'points', 'rebounds'],
 'data': [['A', 5, 11], ['A', 7, 8], ['B', 9, 6], ['B', 12, 6], ['C', 9, 5]]}

Exemplo 5: Converter um DataFrame em um dicionário (“registros”)

O código a seguir mostra como converter um DataFrame do pandas em um dicionário usando o método ‘ registros ‘:

 df. to_dict (' records ')

[{'team': 'A', 'points': 5, 'rebounds': 11},
 {'team': 'A', 'points': 7, 'rebounds': 8},
 {'team': 'B', 'points': 9, 'rebounds': 6},
 {'team': 'B', 'points': 12, 'rebounds': 6},
 {'team': 'C', 'points': 9, 'rebounds': 5}]

Exemplo 6: Converter um DataFrame em um dicionário (“índice”)

O código a seguir mostra como converter um DataFrame do pandas em um dicionário usando o método ‘ index ‘:

 df. to_dict (' index ')

{0: {'team': 'A', 'points': 5, 'rebounds': 11},
 1: {'team': 'A', 'points': 7, 'rebounds': 8},
 2: {'team': 'B', 'points': 9, 'rebounds': 6},
 3: {'team': 'B', 'points': 12, 'rebounds': 6},
 4: {'team': 'C', 'points': 9, 'rebounds': 5}}

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras conversões de dados comuns em pandas:

Como converter Pandas DataFrame em array NumPy
Como converter a série Pandas em array NumPy
Como converter Pandas DataFrame em lista