Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีเขียนข้อความเหนือแท่งใน Bar Plot ด้วย Matplotlib

ใช้ ax.bar() สร้างแท่ง แล้ววนลูปผ่าน patches คำนวณตำแหน่งกึ่งกลางด้านบนของแท่ง (get_x() + get_width()/2, get_height()) และวางข้อความด้วย ax.text()

ขั้นตอนการทำงาน

  1. ตั้งค่าขนาดรูปและเลย์เอาต์
  2. เตรียมข้อมูล years, population และตำแหน่ง x
  3. สร้าง figure และ axes
  4. ตั้งค่าป้ายแกนและหัวข้อ
  5. พลอตแท่งด้วย ax.bar()
  6. วนลูป patches วางข้อความด้านบนแท่ง
  7. แสดงผล

ตัวอย่างโค้ด

from matplotlib import pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

years = [1901, 1911, 1921, 1931, 1941, 1951, 1961, 1971, 1981, 1991, 2001, 2011]
population = [237.4, 238.4, 252.09, 251.31, 278.98, 318.66, 361.09, 439.23, 548.16, 683.33, 846.42, 1028.74]
x = np.arange(len(years))
width = 0.35

fig, ax = plt.subplots()
ax.set_ylabel('Population (in million)')
ax.set_title('Years')
ax.set_xticks(x)
ax.set_xticklabels(years)

pps = ax.bar(x - width / 2, population, width, label='population')

for p in pps:
    height = p.get_height()
    ax.text(x=p.get_x() + p.get_width() / 2, y=height + 0.10, s=f"{height}", ha='center')

plt.show()

ผลลัพธ์

วิธีเขียนข้อความเหนือแท่งใน Bar Plot ด้วย Matplotlib