如何使用 Matplotlib 绘制散点趋势线？

要使用 matplotlib 绘制散点趋势线，我们可以使用 polyfit() 和 poly1d() 方法来获取趋势线点。

步骤

  • 设置图形大小并调整子图之间的内边距和周围的内边距。

  • 使用 numpy 创建 x 和 y 数据点。

  • 创建一个图形和一组子图。

  • 使用 numpy 绘制 x 和 y 数据点。

  • 使用 polyfit() 和 poly1d() 方法查找趋势线数据点。

  • 使用 plot() 方法绘制 x 和 p(x) 数据点。

  • 要显示图形，请使用 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.random.rand(100)
y = np.random.rand(100)
fig, ax = plt.subplots()
_ = ax.scatter(x, y, c=x, cmap='plasma')
z = np.polyfit(x, y, 1)
p = np.poly1d(z)
plt.plot(x, p(x), "r-o")

plt.show()

输出

更新于： 04-Jun-2021

6K+ 浏览量

启动你的 职业生涯

完成课程以获得认证

立即开始
广告