Pandy: jak dodać sumy częściowe do tabeli przestawnej

Często możesz chcieć dodać sumy częściowe do tabeli przestawnej pand.

Na szczęście można to łatwo zrobić, korzystając z wbudowanych funkcji pand.

Poniższy przykład pokazuje, jak to zrobić.

Przykład: dodanie sum częściowych do tabeli przestawnej Pandas

Załóżmy, że mamy następującą ramkę danych pand, która zawiera informacje o różnych koszykarzach:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' position ': ['G', 'G', 'F', 'F', 'G', 'F', 'F', 'F'],
                   ' all_star ': ['Y', 'N', 'Y', 'Y', 'N', 'N', 'N', 'Y'],
                   ' points ': [4, 4, 6, 8, 9, 5, 5, 12]})

#view DataFrame
print (df)

  team position all_star points
0 AGY 4
1 AGN 4
2 AFY 6
3 AFY 8
4 BGN 9
5 BFN 5
6 BFN 5
7 BFY 12

Możemy użyć poniższego kodu, aby utworzyć tabelę przestawną w pandach, która wyświetla sumę punktów dla każdej kombinacji team , all_star i pozycji w DataFrame:

 #create pivot table
my_table = pd. pivot_table (df, values=' points ',
                              index=[' team ', ' all_star '],
                              columns=' position ',
                              aggfunc=' sum ')

#view pivot table
print (my_table)

FG position
team all_star           
AN NaN 4.0
     Y 14.0 4.0
BN 10.0 9.0
     Y 12.0 NaN

Załóżmy teraz, że chcemy dodać wiersz sum częściowych , który wyświetla sumę częściową punktów dla każdego zespołu i pozycji.

W tym celu możemy użyć następującej składni:

 #add subtotals row to pivot table
p.d. concat ([
    y. append ( y.sum (). rename ((x, ' Total ')))
    for x, y in my_table. groupby (level= 0 )
]). append ( my_table.sum (). rename ((' Large ', ' Total ')))

	position F G
team all_star		
A N NaN 4.0
               Y 7.0 4.0
           Overall 7.0 8.0
B N 5.0 9.0
               Y 12.0 NaN
           Total 17.0 9.0
Grand Total 24.0 17.0

Mamy teraz dwa wiersze sum częściowych, które pokazują sumę częściową punktów dla każdego zespołu i pozycji, a także wiersz sumy całkowitej, który pokazuje sumę całkowitą dla każdej kolumny.

Uwaga : pełną dokumentację funkcji obrotowej pandy() można znaleźć tutaj .

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje na pandach:

Pandy: Jak utworzyć tabelę przestawną z liczbą wartości
Pandy: jak zastąpić wartości NaN w tabeli przestawnej zerami
Pandy: Jak przekonwertować tabelę przestawną na ramkę danych