Pandas: วิธีสร้าง bar plot จาก groupby
คุณสามารถใช้ไวยากรณ์ต่อไปนี้เพื่อสร้างกราฟแท่งจากฟังก์ชัน GroupBy ในแพนด้า:
#calculate sum of values by group df_groups = df. groupby ([' group_var '])[' values_var ']. sum () #create bar plot by group df_groups. plot (kind=' bar ')
ตัวอย่างต่อไปนี้แสดงวิธีใช้ไวยากรณ์นี้ในทางปฏิบัติ
ตัวอย่าง: สร้างแผนภูมิแท่งจาก GroupBy ใน Pandas
สมมติว่าเรามี DataFrame แพนด้าต่อไปนี้ซึ่งแสดงคะแนนที่ผู้เล่นบาสเกตบอลจากทีมต่างๆ:
import pandas as pd
#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'A',
'B', 'B', 'B', 'B', 'B',
'C', 'C', 'C', 'C', 'C'],
' points ': [12, 29, 34, 14, 10, 11, 7, 36,
34, 22, 41, 40, 45, 36, 38]})
#view first five rows of DataFrame
df. head ()
team points
0 to 12
1 to 29
2 to 34
3 to 14
4 to 10
เราสามารถใช้ไวยากรณ์ต่อไปนี้เพื่อคำนวณผลรวมคะแนนที่แต่ละทีมทำได้ และสร้างแผนภูมิแท่งเพื่อให้เห็นภาพผลรวมของแต่ละทีม:
import matplotlib. pyplot as plt
#calculate sum of points for each team
df. groupby (' team ')[' points ']. sum ()
#create bar plot by group
df_groups. plot (kind=' bar ')
แกน x แสดงชื่อของแต่ละทีม และแกน y แสดงผลรวมคะแนนที่แต่ละทีมทำได้
นอกจากนี้เรายังสามารถใช้โค้ดต่อไปนี้เพื่อทำให้โครงเรื่องดีขึ้นเล็กน้อย:
import matplotlib. pyplot as plt
#calculate sum of points for each team
df_groups = df. groupby ([' team '])[' points ']. sum ()
#create bar plot with custom aesthetics
df_groups. plot (kind=' bar ', title=' Total Points by Team ',
ylabel=' Total Points ' , xlabel=' Team ', figsize=( 10,6 ) )
#rotate x-axis ticks vertically
plt. xticks (rotation= 0 )
หมายเหตุ : คุณสามารถค้นหาเอกสารฉบับเต็มสำหรับฟังก์ชัน GroupBy ได้ที่นี่
แหล่งข้อมูลเพิ่มเติม
บทช่วยสอนต่อไปนี้จะอธิบายวิธีดำเนินการทั่วไปอื่น ๆ ในแพนด้า:
Pandas: วิธีนับค่าที่ไม่ซ้ำตามกลุ่ม
นุ่น: วิธีคำนวณโหมดตามกลุ่ม
นุ่น: วิธีคำนวณความสัมพันธ์ตามกลุ่ม