Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Tạo Pivot Table với nhiều cột - Python Pandas

Chúng ta có thể tạo Pivot Table với nhiều cột. Để tạo Pivot Table, sử dụng pandas.pivot_table để tạo bảng pivot 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 hồ sơ đội:

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 nhiều cột. Đặt nhiều hơn hai cột:

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

Ví dụ hoàn chỉnh

Mã nguồn đầy đủ:

import pandas as pd
# tạo DataFrame với hồ sơ đội
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("\n... Pivot ...")
# nhiều cột
print(pd.pivot_table(dataFrame, index=["Team ID", "Team Name", "Team Rank"]))

Kết quả

Chương trình sẽ tạo ra kết quả sau:

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