Heim >Backend-Entwicklung >Python-Tutorial >Wie gruppiere ich Daten nach Spalten und konvertiere sie in ein Wörterbuch mit Listen in Python?

Wie gruppiere ich Daten nach Spalten und konvertiere sie in ein Wörterbuch mit Listen in Python?

DDD
DDDOriginal
2024-10-29 13:46:29759Durchsuche

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

GroupBy Results to Dictionary of Lists

Sie haben eine Excel-Tabelle mit drei Spalten, Spalte1, Spalte2 und Spalte3. Sie möchten die Daten extrahieren, sie nach Spalte1 gruppieren und ein Wörterbuch mit Listen erstellen, das wie folgt aussieht:

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

Um dieses Ergebnis zu erzielen, können Sie den folgenden Code verwenden:

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()

Die Groupby-Funktion gruppiert die Daten nach Spalte1 und gibt ein Groupby-Objekt zurück. Anschließend wird die Apply-Methode verwendet, um die Listenfunktion auf die Column3-Daten anzuwenden und eine Werteliste für jede Gruppe zu erstellen. Schließlich wird die to_dict-Methode verwendet, um die gruppierten Daten in ein Wörterbuch mit Listen umzuwandeln.

Das resultierende Wörterbuch hat das von Ihnen angegebene Format, wobei die Schlüssel die Werte in Spalte1 und die Werte Listen der sind entsprechende Werte in Spalte3.

Das obige ist der detaillierte Inhalt vonWie gruppiere ich Daten nach Spalten und konvertiere sie in ein Wörterbuch mit Listen in Python?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!

Stellungnahme:
Der Inhalt dieses Artikels wird freiwillig von Internetnutzern beigesteuert und das Urheberrecht liegt beim ursprünglichen Autor. Diese Website übernimmt keine entsprechende rechtliche Verantwortung. Wenn Sie Inhalte finden, bei denen der Verdacht eines Plagiats oder einer Rechtsverletzung besteht, wenden Sie sich bitte an admin@php.cn