python - 更改 imshow 或类似函数的单元格大小/宽度

Question

我需要第一个和最后一个单元格的宽度为一半。

我的目标是得到这样的东西:

但是我得到了这个:

我的代码:

import numpy as np
import matplotlib.pyplot as plt

data = np.array([0.    , 0.2   , 0.4   , 0.6   , 0.8   , 1.    ])
fig, ax = plt.subplots()
matrix = data.reshape(1, -1)
ax.imshow(matrix, cmap='hot')
plt.show()

有办法做到这一点吗？

Answer 1

正如乔迪评论的那样，如果您只想剪切最外面的单元格，一个快速的快捷方式是更改轴限制:

ax.set_xlim(0, 5)

---

但在更一般的情况下，请使用 pcolormesh用于自定义网格。定义x和y单元边界并在该自定义网格上绘制矩阵:

import numpy as np
import matplotlib.pyplot as plt

data = np.linspace(0, 1, 6)
matrix = data.reshape(1, -1)

# define cell bounds
x = [0, 0.5, 1.5, 2.5, 3.5, 4.5, 5]
y = [-0.5, 0.5]

# plot matrix on custom mesh
fig, ax = plt.subplots()
ax.pcolormesh(x, y, matrix, cmap='hot')

# restyle like imshow
ax.set_aspect('equal')
ax.invert_yaxis()

plt.show()

或者采用更具编程性的方式来定义边界:

r, c = matrix.shape

x = np.arange(c + 1) - 0.5  # [-0.5  0.5  1.5  2.5  3.5  4.5  5.5]
x[0] += 0.5                 # [ 0.   0.5  1.5  2.5  3.5  4.5  5.5]
x[-1] -= 0.5                # [ 0.   0.5  1.5  2.5  3.5  4.5  5. ]

y = np.arange(r + 1) - 0.5  # [-0.5  0.5]