如何在 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()

输出


相关文章