如何强制热图单元格为矩形

Question

我正在使用 pcolormesh 创建热图。

heatmap = plt.pcolormesh(
    grid,
    edgecolors="k",
    cmap=colors.ListedColormap(
        [
            "white",
            "red",
            "blue",
            "green",
            "orange",
            "black",
            "purple",
            "yellow",
            "brown",
            "violet",
            "gray",
        ]
    ),
    linewidth=2,
)
ax = plt.gca()
ax.set_aspect("auto")

我想将单元格形状设置为矩形。我怎样才能做到这一点？

Answer 1

我想你的意思是“方形”单元，而不是“矩形”：

将宽高比设置为 1：

ax.set_aspect(1)

aspect 定义 Y/X 比率，因此单元格高度是两倍：

ax.set_aspect(2)

两倍宽：

ax.set_aspect(1/2)