Heim >Backend-Entwicklung >Python-Tutorial >Wie gruppiere ich Pandas DataFrame-Zeilen und konvertiere Spaltenwerte in Listen?

Wie gruppiere ich Pandas DataFrame-Zeilen und konvertiere Spaltenwerte in Listen?

Linda Hamilton
Linda HamiltonOriginal
2024-12-20 09:13:09697Durchsuche

How to Group Pandas DataFrame Rows and Convert Column Values to Lists?

So konvertieren Sie Datenrahmenzeilen in Listen in Pandas GroupBy

Beim Bearbeiten von Datenrahmen in Pandas kann es erforderlich sein, Daten in eine bestimmte umzuwandeln Format zur weiteren Analyse. Eine Möglichkeit, dies zu tun, besteht darin, Zeilen nach einer bestimmten Spalte zu gruppieren und Listen aus einer anderen Spalte innerhalb jeder Gruppe zu erstellen.

In diesem Szenario erhalten wir einen Datenrahmen mit zwei Spalten: „a“ (Spaltenname) und 'b' (Spaltenwerte). Die Aufgabe besteht darin, diesen Datenrahmen in einen neuen Datenrahmen umzuwandeln, in dem für jeden eindeutigen Wert in Spalte „a“ die entsprechenden Werte aus Spalte „b“ in einer Liste gruppiert sind.

Um dies zu erreichen:

df1 = df.groupby('a')['b'].apply(list).reset_index(name='new')

In diesem Code:

  • df.groupby('a'): Gruppiert den Datenrahmen nach der Spalte 'a'.
  • ['b'].apply(list): Wendet die Listenfunktion auf jede Gruppe an und konvertiert die Werte der Spalte „b“ in eine Liste.
  • reset_index(name= 'neu'): Setzt den Index des resultierenden Datenrahmens zurück und setzt den Namen der neuen Spalte auf 'neu'.

Das Endergebnis ist ein neuer Datenrahmen. df1, mit den eindeutigen Werten aus Spalte „a“ in der Spalte „a“ und den entsprechenden Listen aus Spalte „b“ in der Spalte „neu“.

Hier ist ein Beispiel zur Veranschaulichung:

Angesichts des folgenden Datenrahmens:

a b
A 1
A 2
B 5
B 5
B 4
C 6

Durch Anwenden des oben genannten Codes wird dieser umgewandelt in:

a new
A [1, 2]
B [5, 5, 4]
C [6]

Das obige ist der detaillierte Inhalt vonWie gruppiere ich Pandas DataFrame-Zeilen und konvertiere Spaltenwerte in Listen?. 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