如何在 MatPlotLib 中为散点图绘制平均线?


要在 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()

输出

更新于: 2021 年 6 月 3

3k+ 浏览

职业启航

通过完成课程获得认证

开始
广告
© . All rights reserved.