Heim > Backend-Entwicklung > Python-Tutorial > Wie füge ich gruppierten Daten in Pandas effizient eine sequentielle Zählerspalte hinzu?

Wie füge ich gruppierten Daten in Pandas effizient eine sequentielle Zählerspalte hinzu?

Linda Hamilton
Freigeben: 2024-12-24 14:04:15
Original
902 Leute haben es durchsucht

How to Efficiently Add a Sequential Counter Column to Grouped Data in Pandas?

Sequentielle Zählerspalte zu Gruppen in Pandas DataFrame hinzufügen

Im Zusammenhang mit der Datenverarbeitung gibt es eine häufige Aufgabe, einen sequentiellen Zähler hinzuzufügen Spalte zu Gruppen innerhalb eines Pandas-Datenrahmens. Ein Ansatz besteht darin, eine Callback-Funktion zu verwenden, wie Sie es demonstriert haben:

def callback(x):
    x['seq'] = range(1, x.shape[0] + 1)
    return x
Nach dem Login kopieren

Diese Methode funktioniert zwar, kann aber umständlich sein und erfordert die Definition einer separaten Funktion. Eine prägnantere und elegantere Lösung ist die Verwendung der Methode cumcount():

df.groupby(['c1', 'c2']).cumcount()
Nach dem Login kopieren

Diese Methode berechnet die kumulative Anzahl für jede Gruppe und weist effektiv jeder Zeile innerhalb einer Gruppe eine fortlaufende Nummer zu. Betrachten Sie beispielsweise den folgenden Datenrahmen:

index c1 c2 v1
0 A X 3
1 A X 5
2 A Y 7
3 A Y 1
4 B X 3
5 B X 1
6 B X 3
7 B Y 1
8 C X 7
9 C Y 4
10 C Y 1
11 C Y 6

Die Anwendung von cumcount() auf diesen Datenrahmen, gruppiert nach c1 und c2, würde Folgendes ergeben:

index c1 c2 v1 seq
0 A X 3 1
1 A X 5 2
2 A Y 7 1
3 A Y 1 2
4 B X 3 1
5 B X 1 2
6 B X 3 3
7 B Y 1 1
8 C X 7 1
9 C Y 4 1
10 C Y 1 2
11 C Y 6 3

Die Reihenfolge wird stattdessen bei 1 begonnen von 0, fügen Sie einfach 1 zum cumcount()-Ergebnis hinzu:

df.groupby(['c1', 'c2']).cumcount() + 1
Nach dem Login kopieren

Das obige ist der detaillierte Inhalt vonWie füge ich gruppierten Daten in Pandas effizient eine sequentielle Zählerspalte hinzu?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!

Quelle:php.cn
Erklärung dieser Website
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
Neueste Artikel des Autors
Beliebte Tutorials
Mehr>
Neueste Downloads
Mehr>
Web-Effekte
Quellcode der Website
Website-Materialien
Frontend-Vorlage