Color coded table plot based on the values in a pandas dataframe

Question

I have a dataframe df containing string values

a b c d

b c d a

I would like to produce a pdf plot based on the data in the df, with 4 cols and 2 rows, where each cell in the table plot has a color depending on the value in the df, a=blue, b=red, c=yellow, d=green.

Like this

Thanks in advance!

By "table" you mean a plt.table or a subplot grid? I think you would benefit from describing much more in detail the desired output and also what other resources have not helped you achieve what you want. — ImportanceOfBeingErnest
– ImportanceOfBeingErnest, Commented Nov 13, 2018 at 13:20
@ImportanceOfBeingErnest I added a picture of what I am looking for :) — peter_parker
– peter_parker, Commented Nov 13, 2018 at 13:36
I would probably map the characters to numbers first, then plot an imshow plot of the data. You will need a custom colormap with the 4 colors in it. — ImportanceOfBeingErnest
– ImportanceOfBeingErnest, Commented Nov 13, 2018 at 13:40

Joe · Accepted Answer · 2018-11-13 14:43:22Z

1

You can do in this way:

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

And If you want to remove the ticks, add plt.xticks([]) and plt.yticks([])

edited Nov 13, 2018 at 14:43

answered Nov 13, 2018 at 14:27

Joe

12.4k7 gold badges44 silver badges58 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Color coded table plot based on the values in a pandas dataframe

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related