将热图的单元格分成多行

Question

我正在使用热图，因为每个单元格有 3 行数据，现在，我想将每个单元格分成 3 行，每行数据一个，

这样，每一行都会根据值有自己的颜色

我尝试使用以下 link，但我没有成功分成 3 行：

出于这个原因，我去这个 space 寻求帮助，以便能够进行此修改，我包括我修改过的代码，它属于我提到的 link之前，

from matplotlib import pyplot as plt
import numpy as np

M, N = 4,4
values = np.random.uniform(9, 10, (N * 1, M * 2))

fig, ax = plt.subplots()
#ax.imshow(values, extent=[-0.5, M - 0.5, N - 0.5,-0.5], cmap='autumn_r')
ax.imshow(values, extent=[-0.5,M - 0.5, N - 0.5,-0.5], cmap='autumn_r')

ax.set_xticks(np.arange(0, 4))
ax.set_xticks(np.arange(-0.5, M), minor=True)
ax.set_yticks(np.arange(0, 4))
ax.set_yticks(np.arange(-0.5, N), minor=True)
ax.grid(which='minor', lw=6, color='w', clip_on=True)
ax.grid(which='major', lw=2, color='w', clip_on=True)
ax.tick_params(length=0)
for s in ax.spines:
    ax.spines[s].set_visible(True)
plt.show()

感谢所有的帮助，问候！

Answer 1

将单元格一分为二时，主要的刻度位置既可以用来设置标签，也可以用来定位细分线。要划分为 3 个或更多，明确地绘制水平和垂直线可能更容易。

下面是一些示例代码：

from matplotlib import pyplot as plt
import numpy as np

M, N = 4, 4  # M columns and N rows of large cells
K, L = 1, 3  # K columns and L rows to subdivide each of the cells
values = np.random.uniform(9, 10, (N * L, M * K))

fig, ax = plt.subplots()
ax.imshow(values, extent=[-0.5, M - 0.5, N - 0.5, -0.5], cmap='autumn_r')

# positions for the labels
ax.set_xticks(np.arange(0, M))
ax.set_yticks(np.arange(0, N))

# thin lines between the sub cells
for i in range(M):
    for j in range(1, K):
        ax.axvline(i - 0.5 + j / K, color='white', lw=2)
for i in range(N):
    for j in range(1, L):
        ax.axhline(i - 0.5 + j / L, color='white', lw=2)
# thick line between the large cells
# use clip_on=False and hide the spines to avoid that the border cells look different
for i in range(M + 1):
    ax.axvline(i - 0.5, color='skyblue', lw=4, clip_on=False)
for i in range(N + 1):
    ax.axhline(i - 0.5, color='skyblue', lw=4, clip_on=False)
ax.tick_params(length=0)
for s in ax.spines:
    ax.spines[s].set_visible(False)
plt.show()

将热图的单元格分成多行

Divide cell of heatmap in multiple rows

python

data-visualization

matplotlib

heatmap