Changing The Color Of Matplotlib's Violin Plots
Is there a way to change the color of the violin plots in matplotlib? The default color is this 'brownish' color, which is not too bad, but I'd like to color e.g., the first 3 viol
Solution 1:
matplotlib.pyplot.violinplot()
says it returns:
A dictionary mapping each component of the violinplot to a list of the corresponding collection instances created. The dictionary has the following keys:
bodies
: A list of thematplotlib.collections.PolyCollection
instances containing the filled area of each violin.- [...among others...]
Methods of PolyCollection
s include:
set_color(c)
which sets both the facecolor and edgecolor,set_facecolor(c)
andset_edgecolor(c)
all of which take a "matplotlib color arg or sequence of rgba tuples"
So, it looks like you could just loop through the result's body list and modify the facecolor of each:
violin_parts = plt.violinplot(...)
forpcin violin_parts['bodies']:
pc.set_facecolor('red')
pc.set_edgecolor('black')
It is a bit strange though that you can't set this when creating it like the common plot types. I'd guess it's probably because the operation creates so many bits (the aforementioned PolyCollection
along with 5 other LineCollections
), that additional arguments would be ambiguous.
Solution 2:
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
rrred = '#ff2222'
bluuu = '#2222ff'
x = np.arange(2, 25)
y = np.array([xi * np.random.uniform(0, 1, 10**3) for xi in x]).T
# Create violin plot objects:
fig, ax = plt.subplots(1, 1, figsize = (8,8))
violin_parts = ax.violinplot(y, x, widths = 0.9, showmeans = True, showextrema = True, showmedians = True)
# Make all the violin statistics marks red:for partname in ('cbars','cmins','cmaxes','cmeans','cmedians'):
vp = violin_parts[partname]
vp.set_edgecolor(rrred)
vp.set_linewidth(1)
# Make the violin body blue with a red border:for vp in violin_parts['bodies']:
vp.set_facecolor(bluuu)
vp.set_edgecolor(rrred)
vp.set_linewidth(1)
vp.set_alpha(0.5)
Post a Comment for "Changing The Color Of Matplotlib's Violin Plots"