Cách gán nhãn cho biểu đồ bong bóng/biểu đồ phân tán từ cột DataFrame Pandas

Giới thiệu

Để gán nhãn cho biểu đồ bong bóng hoặc biểu đồ phân tán (scatter plot) bằng dữ liệu từ một cột trong Pandas DataFrame, bạn có thể làm theo các bước sau:

• Thiết lập kích thước hình vẽ và điều chỉnh khoảng cách padding.
• Tạo một DataFrame df chứa dữ liệu hai chiều.
• Tạo biểu đồ phân tán với df.
• Ghi chú (annotate) từng điểm dữ liệu bằng văn bản.
• Hiển thị hình vẽ bằng phương thức show().

Ví dụ minh họa

import pandas as pd
from matplotlib import pyplot as plt

# Thiết lập kích thước hình
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

# Tạo DataFrame
df = pd.DataFrame(
    dict(
        x=[1, 3, 2, 4, 5],
        y=[0, 3, 1, 2, 5],
        points=['Kiran', 'Raju', 'Raja', 'Javed', 'Rishi']
    )
)

# Vẽ scatter plot
ax = df.plot.scatter(x='x', y='y', alpha=0.5)

# Ghi chú từng điểm dữ liệu
for i, txt in enumerate(df.points):
    ax.annotate(txt, (df.x.iat[i] + 0.05, df.y.iat[i]))

plt.show()

Kết quả đầu ra

Chương trình sẽ tạo ra biểu đồ phân tán có gán nhãn: