在python中绘制彩色网格

Question

我正在尝试在python中创建2D绘图，其中水平轴分为多个间隔或列，每列的颜色沿垂直轴变化。

每个间隔的颜色取决于时间的周期性函数的值。 为了简单起见，假设这些值在0到1之间。然后，接近1的值应该是深红色，接近0的值应该是深蓝色（实际颜色并不重要）。

这是情节外观的示例：

有没有办法在Python中使用matplotlib做到这一点？

Answer 1

这实际上只是显示图像。 您可以使用imshow做到这imshow 。

import matplotlib.pyplot as plt

import numpy as np

# Just some example data (random)
data = np.random.rand(10,5)

rows,cols = data.shape

plt.imshow(data, interpolation='nearest', 
                 extent=[0.5, 0.5+cols, 0.5, 0.5+rows],
                 cmap='bwr')