颜色条按示例matplotlib

Question

I am trying to color code a bar plot I have by values in a list. I am plotting something like this but I can't figure out how to color it:

x=[1,2,3]
y=[1,2,3]
type=[big,small,big]

p1=plt.bar(x,y)
plt.colorbar(type)

Answer 1

So what I ended up doing was creating a dictionary with defined colors for each of my parameters and then just converted this when plotting like so:

colors = {'C':'b', 'T':'r', 'G':'g', 'A':'y'}
wts.append(colors[x[3]])
p1 = plt.bar(loci, vafs, color=wts)

Answer 2

You may create a dictionary mapping values from the typ list to colors. Then using this dictionary you can create a list of colors to supply to the bar 's color argument.

In order to create a legend, the same dictionary can be used to set the color of legend handles.

import matplotlib.pyplot as plt

x=[1,2,3]
y=[1,2,3]
typ=["big","small","big"]
color = {"big" : "indigo", "small" : "mediumvioletred"}
p1 = plt.bar(x,y, color=[color[t] for t in typ])

handles = [plt.Rectangle((0,0),1,1, color=color[c]) for c in set(typ)]
plt.legend(handles=handles, labels=set(typ))

plt.show()