python - 基于 pandas 数据框中的值的颜色编码表格图

Question

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

a b c d

bcda

我想根据 df 中的数据生成一个 pdf 图，有 4 列和 2 行，其中表图中的每个单元格的颜色取决于 df 中的值，a=蓝色，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()

如果您想删除刻度，请添加plt.xticks([])和plt.yticks([])