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

hs1ihplo  于 2023-04-10  发布在  Python
关注(0)|答案(1)|浏览(231)

我需要第一个和最后一个单元格是宽度的一半。
我的目标是得到这样的东西:

但我得到了这个

我的代码:

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()

是否有选项可以执行此操作?

ztyzrc3y

ztyzrc3y1#

正如Jody所评论的,如果你只想裁剪最外面的单元格,一个快捷方式是更改轴限制:

ax.set_xlim(0, 5)

但在更一般的情况下,使用pcolormesh来自定义网格。定义xy单元格边界,并在该自定义网格上绘制matrix

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]

相关问题