在matplotlib中使用矩形插值绘制一条线

3

我想绘制数据,其中连续的点由矩形的部分连接,可以是先水平再垂直,也可以是先垂直再水平。以下是一种朴素的方法:

import matplotlib.pyplot as plt

x_data = [0.0, 1.0, 3.0, 4.5, 7.0]
y_data = [1.5, 3.5, 6.0, 2.0, 9.0]

# Linear interpolation
plt.plot(x_data, y_data, label='linear_interp')

# Vertical first interpolation
x_data_vert_first = [0.0, 0.0, 1.0, 1.0, 3.0, 3.0, 4.5, 4.5, 7.0]
y_data_vert_first = [1.5, 3.5, 3.5, 6.0, 6.0, 2.0, 2.0, 9.0, 9.0]
plt.plot(x_data_vert_first, y_data_vert_first, label="vert_first")

# Horizontal first interpolation
x_data_flat_first = [0.0, 1.0, 1.0, 3.0, 3.0, 4.5, 4.5, 7.0, 7.0]
y_data_flat_first = [1.5, 1.5, 3.5, 3.5, 6.0, 6.0, 2.0, 2.0, 9.0]
plt.plot(x_data_flat_first, y_data_flat_first, label="flat_first")

plt.legend(loc='upper left')
plt.show()

插值比较

是否有pyplot选项可以实现这一点?Numpy或Scipy中内置的插值功能?我在文档中没有看到任何相关内容(例如,这不是箱形图或条形图,而是不同的东西)

我可以编写一个简单的函数来执行此类插值,但如果可能的话,我宁愿使用库。

1个回答

6
你可以使用 plt.step 来获得相同的结果:
plt.plot(x_data, y_data, label='linear_interp')
plt.step(x_data, y_data, where = 'pre', label = 'vert_first')
plt.step(x_data, y_data, where = 'post', label = 'flat_first')
plt.legend(loc='upper left')
plt.show()

那就是我所缺少的,谢谢! - ijustlovemath

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接