Heim >Backend-Entwicklung >Python-Tutorial >Wie gruppiere ich Daten nach Spalten und konvertiere sie in ein Wörterbuch mit Listen 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!