python - 具有指定轴的颜色图

Question

我有一个包含时间点的向量、一个包含频率点(对数尺度)的向量和一个 (M,N) 矩阵，其中 M 是频率数，N 是时间点数。

我想绘制一个图，其中 x 轴是时间向量，y 轴是频率向量(以对数刻度表示)，每个点都有一个由矩阵中相应值确定的颜色。

我一直在看matplotlib的imshow，但我似乎无法让它工作，也找不到一个很好的例子。

Answer 1

听起来您想要 pcolormesh 而不是 imshow。

例如:

import numpy as np
import matplotlib.pyplot as plt

m, n = 20, 30

# Generate randomly spaced, but increasing time and frequency vectors
time = np.random.normal(0, 1, n).cumsum()
frequency = np.random.normal(0, 1, m).cumsum()
data = np.random.random((m,n))

fig, ax = plt.subplots()
ax.pcolormesh(time, frequency, data, cmap='gist_earth')
ax.axis('tight')
plt.show()