在python中绘制彩色网格

Question

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

每个区间的颜色取决于周期性时间函数的值。为简单起见，我们假设这些值介于0和1之间。然后，接近1的值应为深红色，接近0的值应为深蓝色（例如，实际颜色并不重要）。

以下是情节应如下所示的示例：

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

Answer 1

这实际上只是显示图像。您可以使用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')