如何在 MatPlotLib 中为散点图绘制平均线?
matplotlibpythondata visualization
要在 matplotlib 中为图绘制平均线,我们可以采取以下步骤 −
设置图形大小并调整子图之间和周围的填充。
使用 numpy 制作 x 和 y 数据点。
使用 subplots() 方法创建一个图形和一组子图。
对 x 和 y 数据点使用 plot() 方法。
查找数组 x 的平均值。
使用 plot() 方法绘制 x 和 y_avg 数据点。
在图形。
要显示图形,请使用 show() 方法。
示例
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()