Để vẽ đường ngang đại diện cho giá trị trung bình của dữ liệu trên scatter plot:
Các bước
- Thiết lập figure size và autolayout.
- Tạo mảng x, y bằng NumPy.
- Tạo figure, subplots và vẽ scatter/line plot.
- Tính trung bình của x: np.mean(x), tạo mảng y_avg cùng độ dài x.
- Vẽ đường trung bình kiểu nét đứt, màu đỏ, độ dày lớn.
- Thêm legend và hiển thị bằng show().
Ví dụ
import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
x = np.array([3, 4, 5, 6, 7, 8, 9])
y = np.array([6, 5, 4, 3, 2, 1, 6])
fig, ax = plt.subplots()
ax.plot(x, y, 'o-', label='line plot')
y_avg = [np.mean(x)] * len(x)
ax.plot(x, y_avg, color='red', lw=6, ls='--', label="average plot")
plt.legend(loc=0)
plt.show()
Kết quả
