如何从 Matplotlib 图中提取数据?

matplotlibpythondata visualization

要从 matplotlib 中的图中提取数据,我们可以使用 get_xdata()get_ydata() 方法。

步骤

  • 设置图形大小并调整子图之间和周围的填充。
  • 使用 numpy 创建 y 数据点。
  • 使用 color=redlinewidth=5 绘制 y 数据点。
  • 打印数据提取语句。
  • 使用 get_xdata()get_ydata() 方法从图中提取数据(步骤 3)。
  • 打印 x 和 y 数据(步骤 5)。
  • 要显示图形,使用 show() 方法。

示例

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
y = np.array([1, 3, 2, 5, 2, 3, 1])
curve, = plt.plot(y, c='red', lw=5)
print("从绘图中提取数据....")
xdata = curve.get_xdata()
ydata = curve.get_ydata()
print("图的 X 数据点是:", xdata)
print("绘图的 Y 数据点为:", ydata)
plt.show()

输出

从绘图中提取数据....
绘图的 X 数据点为:[0. 1. 2. 3. 4. 5. 6.]
绘图的 Y 数据点为:[1 3 2 5 2 3 1]


相关文章