python - 将小提琴图中的均值指示器更改为圆形

Question

我想改变 violinplots 中平均值的外观。我正在使用matplotlib。我可以使用以下代码更改手段的颜色：

 import matplotlib.pyplot as plt

 fig,(axes1,axes2,axes3) = plt.subplots(nrows=3,ncols=1,figsize=(10,20))

 r=axes2.violinplot(D,showmeans=True,showmedians=True)
 r['cmeans'].set_color('red')

但现在我想将平均值（目前是一条线，如中位数）的外观更改为“小圆圈”。有人可以帮我弄这个吗？

Answer 1

这个想法可以是获取平均线的坐标并在这些坐标处绘制散点图。

获取坐标可以

• 要么通过循环平均线的路径来完成，

  # loop over the paths of the mean lines
  xy = [[l.vertices[:,0].mean(),l.vertices[0,1]] for l in r['cmeans'].get_paths()]
  xy = np.array(xy)
  

• 或通过重新计算输入数据的平均值。

  #alternatively get the means from the data
  y = data.mean(axis=0)
  x = np.arange(1,len(y)+1)
  xy=np.c_[x,y] 
  

完整代码：

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(1)

data = np.random.normal(size=(50, 2))

fig,ax = plt.subplots()

r=ax.violinplot(data,showmeans=True)

# loop over the paths of the mean lines
xy = [[l.vertices[:,0].mean(),l.vertices[0,1]] for l in r['cmeans'].get_paths()]
xy = np.array(xy)
##alternatively get the means from the data
#y = data.mean(axis=0)
#x = np.arange(1,len(y)+1)
#xy=np.c_[x,y] 

ax.scatter(xy[:,0], xy[:,1],s=121, c="crimson", marker="o", zorder=3)

# make lines invisible
r['cmeans'].set_visible(False)

plt.show()