Pandas: як змінити назви стовпців у зведеній таблиці

Часто вам може знадобитися змінити або відформатувати назви стовпців у зведеній таблиці pandas певним чином.

На щастя, це легко зробити за допомогою вбудованих функцій pandas.

У наступному прикладі показано, як це зробити.

Приклад: змініть назви стовпців у зведеній таблиці Pandas

Припустімо, що у нас є такий фрейм даних pandas, який містить інформацію про різних баскетболістів:

 import pandas as pd

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

#view DataFrame
print (df)

	team position points
0 A G 4
1 A G 4
2 A F 6
3 A C 8
4 B G 9
5 B C 5
6 B F 5
7 B F 12

Ми можемо використати наступний код, щоб створити зведену таблицю в pandas, яка показуватиме середнє значення балів для кожної команди та позиції в DataFrame:

 #create pivot table
piv = pd. pivot_table (df, values=' points ', index=' team ', columns=' position ')

#view pivot table
print (piv)

CFG position
team                   
A 8.0 6.0 4.0
B 5.0 8.5 9.0

Тепер припустімо, що ми хочемо видалити позицію слова у зведеній таблиці та видалити додатковий рядок команди зі зведеної таблиці.

Для цього ми можемо використати такий синтаксис:

 #format column names
iv. columns = ['_'. join (str(s). strip () for s in col if s) for col in piv. columns ]

#reset indexes
iv. reset_index (inplace= True )

#view updated pivot table
print (piv)

  team CFG
0 A 8.0 6.0 4.0
1 B 5.0 8.5 9.0

Зверніть увагу, що ми змогли видалити позицію слова у зведеній таблиці та видалити додатковий рядок команди зі зведеної таблиці.

Також зауважте, що це загальне рішення також працюватиме для зведеної таблиці з мультиіндексом.

Примітка . Ви можете знайти повну документацію щодо функції pandas pivot_table() тут .

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в pandas:

Pandas: Як створити зведену таблицю з кількістю значень
Pandas: як замінити значення NaN у зведеній таблиці на нулі
Pandas: Як перетворити PivotTable на DataFrame