Python小提琴情节

时间:2015-04-23 03:04:57

标签: python matplotlib plot

我如何使用ggplot2在R中做小提琴情节?在ggplot2中你可以制作出漂亮的小提琴情节,如:

ggplot(dat,aes(value,prob)) + 
geom_violin() + 
geom_jitter(alpha=0.1) + 
scale_y_log10() +
labs(x="test values", y="prob")

你可以获得这样的好图:enter image description here

我怎样才能在Python中做同样的事情?我看过matplotlib.pyplot.violinplot但是却无法得到类似的东西。

1 个答案:

答案 0 :(得分:2)

使用seaborn(注意这是使用开发版本;当前版本中的工作方式稍有不同,但确实做了小提琴):

sns.violinplot(x=value, y=prob, inner=None, color="white", cut=0)
sns.stripplot(x=value, y=prob, jitter=.3,  color="black", alpha=.1, size=4)

enter image description here

这些函数都返回matplotlib轴对象,您可以对该对象执行ax.set_yscale("log")以使用日志轴。请注意,这会在符合KDE之后设置的比例(我不确定ggplot会做什么)。