Cách viết text trên đầu cột trong biểu đồ cột (Bar Plot) Python Matplotlib

Để viết text trên đầu các cột trong bar plot, ta thực hiện các bước sau:

Các bước thực hiện

• Thiết lập kích thước hình và khoảng đệm.
• Tạo danh sách year, population và x. Khởi tạo biến width.
• Tạo hình và subplots bằng subplots().
• Đặt ylabel, title, xticks và xticklabels.
• Vẽ cột bằng bar() với x, population, width.
• Lặp qua các patch cột, đặt text ở đỉnh cột bằng text().
• Hiển thị bằng show().

Ví dụ

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 + .10, s="{}".format(height), ha='center')

plt.show()

Kết quả