matplotlib imshow -- 使用任何向量作为轴

Question

我想使用任何向量作为plt.imshow().

A = np.random.rand(4, 4)
x = np.array([1, 2, 3, 8])
y = np.array([-1, 0, 2, 3])

我想象这样的事情：

plt.imshow(a, x_ax=x, y_ax=y)

我知道有一个extent可用的参数，但遗憾的是它不允许非等距向量。

有人可以帮忙吗？提前致谢。

Answer 1

Imshow 图总是等距分布。问题是您是否想要
(a) 具有不等距标签的等距图，或
(b) 具有可缩放标签的不等距图。

(a) 等距图

import numpy as np
import matplotlib.pyplot as plt

a = np.random.rand(4, 4)
x = np.array([1, 2, 3, 8])
y = np.array([-1, 0, 2, 3])

plt.imshow(a)
plt.xticks(range(len(x)), x)
plt.yticks(range(len(y)), y)
plt.show()

(b) 不等距图

import numpy as np
import matplotlib.pyplot as plt

a = np.random.rand(3, 3)
x = np.array([1, 2, 3, 8])
y = np.array([-1, 0, 2, 3])
X,Y = np.meshgrid(x,y)

plt.pcolormesh(X,Y,a)

plt.xticks(x)
plt.yticks(y)
plt.show()

请注意，在这种情况下，“向量”将指定网格的边缘，因此它们只允许使用 3x3 数组。