根据数组和颜色映射着色matplotlib的背景

Question

我想知道是否可以根据绘制的数据对典型的matplotlib图的背景进行着色。

为简单起见，我们说：

x=arange(1,5,0.01)
y=sin(x)
plot(x,y)

是否可以根据y值对轴的背景进行着色？

可以通过将包含x和y的数组传递给imshow来实现着色，例如：

imshow(array, cmap='hot')

虽然我希望在这个不可思议的数字之上有一个x和y的线图。

请问这可能吗？

Answer 1

当然可以：

x = arange(1,5,0.01)
yarr = vstack((x,))
y = sin(x)

imshow(yarr, extent=(min(x),max(x), min(y),max(y)), cmap=cm.hot)
plot(x, y, color='cornflowerblue',lw=4)

范围关键字将图像的限制与绘制的数据相匹配。

这会给你：