网上很多人使用seaborn
绘制小提琴图,本人暂时不想学新的东西,就是懒。本文介绍如何使用matplotlib绘制小提琴图,很多其他博客只是使用最简单的语法,默认小提琴颜色会是蓝色,根本改不了。本文使用了一点高级的用法,可以满足基本日常需求。
import matplotlib.pyplot as plt
import numpy as np
# plt.rcParams["font.family"] = "SimSun"这个是宋体。
plt.rc('font',family='Times New Roman')
plt.rcParams["axes.unicode_minus"] = False
violin1,violin2,violin3,violin4,violin5,violin6=np.random.randn(6,100)
figure=plt.violinplot( [violin1,violin2,violin3,violin4,violin5,violin6],showextrema=False,showmeans=True#高级知识点1:不显示极值,显示均值。
,positions=[0.7, 1.3, 2.7,3.3,4.7,5.3])#高级知识点2:position参数,控制6个小提琴画在哪里。
color=['C1','C2','C1','C2','C1','C2']
for i, pc in enumerate(figure['bodies']):
pc.set_facecolor(color[i])#高级知识点3:控制各个小提琴的颜色。
pc.set_edgecolor('black')
pc.set_alpha(1)
plt.ylabel('Values')
plt.xlabel('Time')
plt.xticks([1,3,5],[0.5,1,1.5])
plt.title("Value Calculation")
plt.show()
效果如下:
想要更加高级怎么办?这边推荐您前往官网呢,客官~