如何同时绘制带有样式编码和颜色编码的swarmplot / stripplot

时间:2020-11-01 14:00:03

标签: python matplotlib seaborn

我有一些数据想要同时绘制为样式编码和颜色编码。但它们彼此重叠,所以我需要使点成簇或抖动。

seaborn可以选择分散地同时进行样式编码和颜色编码。

tips = sns.load_dataset("tips")
sns.scatterplot(data=tips, x="total_bill", y="tip", hue="day", style="time")

enter image description here

但在swarmplot / stripplot中,仅允许使用色相,但不允许使用样式编码

(顺便说一句,我知道seaborn散点图有“抖动”的说法,但截至2020年11月1日,它尚不可用)

我想知道是否有一种方法可以使数据在样式编码和颜色编码时成群或抖动

谢谢!

1 个答案:

答案 0 :(得分:1)

创建散点图后,可以添加一些抖动:

import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns

tips = sns.load_dataset("tips")
ax = sns.scatterplot(data=tips, x="total_bill", y="tip", hue="day", style="time")
for dots in ax.collections:
    offsets = dots.get_offsets()
    jittered_offsets = offsets + np.random.normal(0, 0.2, offsets.shape)
    dots.set_offsets(jittered_offsets)
plt.show()

example plot