Tạo Pivot Table dưới dạng DataFrame – Python Pandas

Để tạo Pivot Table, sử dụng hàm pandas.pivot_table() để tạo bảng xoay kiểu spreadsheet dưới dạng DataFrame.

Trước tiên, nhập thư viện cần thiết:

import pandas as pd

Tạo DataFrame với bản ghi Team:

dataFrame = pd.DataFrame({
    'Team ID': {0: 5, 1: 9, 2: 6, 3: 11, 4: 2, 5: 7},
    'Team Name': {0: 'India', 1: 'Australia', 2: 'Bangladesh', 3: 'South Africa', 4: 'Sri Lanka', 5: 'England'},
    'Team Points': {0: 95, 1: 93, 2: 42, 3: 60, 4: 80, 5: 55},
    'Team Rank': {0: 'One', 1: 'Two', 2: 'Six', 3: 'Four', 4: 'Three', 5: 'Five'}
})

Tạo Pivot Table với một cột đơn:

pd.pivot_table(dataFrame, index=["Team ID"])

Ví dụ hoàn chỉnh

import pandas as pd

# Tạo DataFrame với bản ghi Team
dataFrame = pd.DataFrame({
    'Team ID': {0: 5, 1: 9, 2: 6, 3: 11, 4: 2, 5: 7},
    'Team Name': {0: 'India', 1: 'Australia', 2: 'Bangladesh', 3: 'South Africa', 4: 'Sri Lanka', 5: 'England'},
    'Team Points': {0: 95, 1: 93, 2: 42, 3: 60, 4: 80, 5: 55},
    'Team Rank': {0: 'One', 1: 'Two', 2: 'Six', 3: 'Four', 4: 'Three', 5: 'Five'}
})

print("DataFrame...\n", dataFrame)
print("\n... Pivot ...")
print(pd.pivot_table(dataFrame, index=["Team ID"]))

Đầu ra

Sẽ tạo ra đầu ra sau:

DataFrame...
   Team ID   Team Name  Team Points Team Rank
0        5       India           95       One
1        9   Australia           93       Two
2        6  Bangladesh           42       Six
3       11 South Africa           60      Four
4        2  Sri Lanka           80     Three
5        7     England           55      Five

... Pivot ...
          Team Points
Team ID
2                80
5                95
6                42
7                55
9                93
11               60