在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')