График таблицы с цветовой кодировкой, основанный на значениях в кадре данных pandas

Question

У меня есть датафрейм df, содержащий строковые значения

abcd

bcda

Я хотел бы создать pdf-график на основе данных в df с 4 столбцамии 2 строки, где каждая ячейка на графике таблицы имеет цвет в зависимости от значения в df, a = синий, b = красный, c = желтый, d = зеленый.

Как это

Заранее спасибо!

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([])