在 Matplotlib 中更改箱线图/小提琴图的轴标签

Question

稍微调整一下这里的代码：http://pyinsci.blogspot.com/2009/09/violin-plot-with-matplotlib.html，如下所示，我可以得到用Python生成的小提琴图，如下所示：

# Import modules
import pylab as pl
from scipy import stats
import numpy as np

# Function for Violin Plot

def violin_plot(ax,data,groups,bp=False):
    '''Create violin plot along an axis'''
    dist = max(groups) - min(groups)
    w = min(0.15*max(dist,1.0),0.5)
    for d,p in zip(data,groups):
        k = stats.gaussian_kde(d) #calculates the kernel density
        m = k.dataset.min() #lower bound of violin
        M = k.dataset.max() #upper bound of violin
        x = np.arange(m,M,(M-m)/100.) # support for violin
        v = k.evaluate(x) #violin profile (density curve)
        v = v/v.max()*w #scaling the violin to the available space
        ax.fill_betweenx(x,p,v+p,facecolor='y',alpha=0.3)
        ax.fill_betweenx(x,p,-v+p,facecolor='y',alpha=0.3)
    if bp:
        ax.boxplot(data,notch=1,positions=pos,vert=1)

groups = range(3)
a = np.random.normal(size=100)
b = np.random.normal(size=100)
c = np.random.normal(size=100)

data = np.vstack((a,b,c))
fig = pl.figure()
ax = fig.add_subplot(111)
violin_plot(ax,data,groups,bp=0)
pl.show()

这会产生一个像这样的数字：

我想要做的是更改刻度线的标签，这样不是 -0.5 到 2.5 x 0.5 个数字，而是在 0.0 处有一个“A”，在 1.0 处有一个“B”，在 1.0 处有一个“C” 2.0。

是否有捷径可寻？

Answer 1

在致电 之前pl.show，请使用：

ax.set_xticks([0, 1, 2])
ax.set_xticklabels(['A', 'B', 'C'])