基于熊猫数据框中的值的彩色编码表图

Question

我有一个包含字符串值的数据框df

a b c d

b c d a

我想基于df中的数据生成一个pdf图，具有4列和2行，其中表格中每个单元格的颜色取决于df中的值，a = blue，b =红色，c =黄色，d =绿色。

Like this

谢谢！

Answer 1

您可以通过以下方式进行：

from matplotlib import colors as c

color_map = {'a':1,'b':2,'c':3, 'd':4}
cMap = c.ListedColormap(['g','b','y','r'])
df = df.replace(color_map)
fig, ax = plt.subplots()
ax.pcolor(df,cmap=cMap)
plt.show()

如果要删除ticks，请添加plt.xticks([])和plt.yticks([])