在matplotlib.collections中使用路径绘制曲线
发布时间:2023-12-24 02:36:23
在matplotlib中,使用路径绘制曲线带是非常常见的操作之一。路径绘制曲线带可以用来表示数据的变化范围,例如误差范围、置信区间等。在matplotlib中,可以使用matplotlib.collections中的PolyCollection类来绘制路径曲线带。
下面是一个简单的使用例子:
import numpy as np import matplotlib.pyplot as plt from matplotlib.collections import PolyCollection # 生成数据 x = np.linspace(0, 2*np.pi, 100) y_mean = np.sin(x) y_lower = np.sin(x) - 0.2 y_upper = np.sin(x) + 0.2 # 创建路径 verts = list(zip(x, y_lower)) + list(zip(x[::-1], y_upper[::-1])) # 创建PolyCollection对象 poly = PolyCollection([verts], facecolors='blue', alpha=0.3) # 创建图像对象 fig, ax = plt.subplots() # 添加路径对象到图像对象 ax.add_collection(poly) # 绘制均值曲线 ax.plot(x, y_mean, color='blue') # 设置轴范围 ax.set_xlim(0, 2*np.pi) ax.set_ylim(-1.2, 1.2) # 显示图像 plt.show()
在这个例子中,我们使用numpy生成了一组x坐标和对应的y坐标的上界和下界。然后,我们将两组坐标合并为一个列表,作为路径对象的输入。我们创建了一个PolyCollection对象,并设置了路径的颜色和透明度。然后,我们将路径对象添加到图像对象中,并使用plot函数绘制了均值曲线。最后,我们设置了轴的范围,并显示了图像。
在运行这段代码后,我们会得到一个显示了路径曲线带的图像,其中蓝色的线代表了数据的均值,蓝色的区域代表了数据的变化范围。
综上所述,使用路径绘制曲线带是matplotlib中的常见操作之一。通过使用matplotlib.collections中的PolyCollection类,我们可以方便地绘制路径曲线带,并将其添加到图像中。
