如何使用 matplotlib 和 seaborn 制作一个带有百分比的问题矩阵？

Question

我想制作您在下面看到的这种类型的图表。我知道我可以用 matplotlib 制作矩阵图像这样

    cmap = colors.ListedColormap(['white','red'])
data = [
    [0,0,0,0,0,1,1,1,1,],
    [0,0,0,0,0,1,0,0,1,],
]
plt.figure(figsize=(9,5))
plt.pcolor(data[::-1],cmap=cmap,edgecolors='k', linewidths=3)
plt.xlabel('Problem') 
plt.ylabel('Particpant')
plt.show()

但是我将如何添加要包含在此图表中的百分比？

Answer 1

您可以添加辅助 x-axis (ax.twiny())，使用顶部轴进行编号，底部轴显示百分比。

使用偏移 0.5 的 x 和 y 位置列表调用 pcolor 会将刻度和刻度标签置于整数位置。 clip_on=False 确保外部单元格边框的厚度与其余部分相同。 ax.invert_yaxis() 让您反转 y 轴（因此您可以使用 data 而不是 data[::-1]）。

import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
import numpy as np

cmap = ListedColormap(['white', 'orangered'])
data = np.random.randint(0, 3, size=(28, 30)) % 2
data[:, 9] = 1  # one full column to simulate 100%
data[:, 11] = 0  # one empty  column to simulate 0%
fig, ax = plt.subplots(figsize=(9, 5))
ax.pcolor(np.arange(data.shape[1] + 1) + 0.5, np.arange(data.shape[0] + 1) + 0.5, data,
          cmap=cmap, edgecolors='k', linewidths=3, clip_on=False)
ax.set_yticks(range(1, data.shape[0] + 1))
ax.set_xticks(range(1, data.shape[1] + 1))
ax.set_xticklabels([f'{p:.0f}' for p in data.mean(axis=0) * 100])
ax.invert_yaxis()

ax2 = ax.twiny()
ax2.set_xlim(ax.get_xlim())
ax2.set_xticks(range(1, data.shape[1] + 1))
ax2.set_xlabel('Problem')
ax.tick_params(length=0)
ax2.tick_params(length=0)
ax.set_ylabel('Particpant')
plt.tight_layout()
plt.show()

减小字体大小（或增大 figsize）也允许显示百分号：

ax.set_xticklabels([f'{p:.0f}%' for p in data.mean(axis=0) * 100], fontsize=8)

如何使用 matplotlib 和 seaborn 制作一个带有百分比的问题矩阵？

How can I make a problem matrix with percentage using matplotlib and seaborn?

python

graph

matplotlib

matrix

seaborn