Konvertieren Sie ein Pandas DataFrame in ein Dictionary

Question

Ich habe ein DataFrame mit vier Spalten. Ich möchte dieses DataFrame in ein Python-Wörterbuch umwandeln. Ich möchte, dass die Elemente der ersten Spalte keys sind und die Elemente der anderen Spalten in der gleichen Zeile values sind.

DataFrame:

    ID   A   B   C
0   p    1   3   2
1   q    4   3   2
2   r    4   0   9 

Die Ausgabe sollte so aussehen:

{'p': [1,3,2], 'q': [4,3,2], 'r': [4,0,9]}

Answer 1

Mit dieser Methode werden die Spalten des Datenrahmens die Schlüssel und die Serien des Datenrahmens die Werte sein.`

data_dict = dict()
for col in dataframe.columns:
    data_dict[col] = dataframe[col].values.tolist()

Answer 2

DataFrame.to_dict() konvertiert DataFrame in ein Wörterbuch.

Beispiel

>>> df = pd.DataFrame(
    {'col1': [1, 2], 'col2': [0.5, 0.75]}, index=['a', 'b'])
>>> df
   col1  col2
a     1   0.1
b     2   0.2
>>> df.to_dict()
{'col1': {'a': 1, 'b': 2}, 'col2': {'a': 0.5, 'b': 0.75}}

Siehe diese Dokumentation für weitere Details