seaborn.swarmplot

优质
小牛编辑
139浏览
2023-12-01
seaborn.swarmplot(x=None, y=None, hue=None, data=None, order=None, hue_order=None, dodge=False, orient=None, color=None, palette=None, size=5, edgecolor='gray', linewidth=0, ax=None, **kwargs)

Draw a categorical scatterplot with non-overlapping points.

This function is similar to stripplot(), but the points are adjusted (only along the categorical axis) so that they don’t overlap. This gives a better representation of the distribution of values, but it does not scale well to large numbers of observations. This style of plot is sometimes called a “beeswarm”.

A swarm plot can be drawn on its own, but it is also a good complement to a box or violin plot in cases where you want to show all observations along with some representation of the underlying distribution.

Arranging the points properly requires an accurate transformation between data and point coordinates. This means that non-default axis limits must be set before drawing the plot.

Input data can be passed in a variety of formats, including:

  • Vectors of data represented as lists, numpy arrays, or pandas Series objects passed directly to the x, y, and/or hue parameters.
  • A “long-form” DataFrame, in which case the x, y, and hue variables will determine how the data are plotted.
  • A “wide-form” DataFrame, such that each numeric column will be plotted.
  • An array or list of vectors.

In most cases, it is possible to use numpy or Python objects, but pandas objects are preferable because the associated names will be used to annotate the axes. Additionally, you can use Categorical types for the grouping variables to control the order of plot elements.

This function always treats one of the variables as categorical and draws data at ordinal positions (0, 1, … n) on the relevant axis, even when the data has a numeric or date type.

See the color_palette(), or a dictionary mapping hue levels to matplotlib colors.

size:float, optional

Diameter of the markers, in points. (Although plt.scatter is used to draw the points, the size argument here takes a “normal” markersize and not size^2 like plt.scatter.

edgecolor:matplotlib color, “gray” is special-cased, optional

Color of the lines around each point. If you pass "gray", the brightness is determined by the color palette used for the body of the points.

linewidth:float, optional

Width of the gray lines that frame the plot elements.

ax:matplotlib Axes, optional

Axes object to draw the plot onto, otherwise uses the current Axes.

返回值:ax:matplotlib Axes

Returns the Axes object with the plot drawn onto it.

See also

A traditional box-and-whisker plot with a similar API.A combination of boxplot and kernel density estimation.A scatterplot where one variable is categorical. Can be used in conjunction with other plots to show each observation.Combine a categorical plot with a class:FacetGrid.

Examples

Draw a single horizontal swarm plot:

>>> import seaborn as sns
>>> sns.set(style="whitegrid")
>>> tips = sns.load_dataset("tips")
>>> ax = sns.swarmplot(x=tips["total_bill"])

http://seaborn.pydata.org/_images/seaborn-swarmplot-1.png

Group the swarms by a categorical variable:

>>> ax = sns.swarmplot(x="day", y="total_bill", data=tips)

http://seaborn.pydata.org/_images/seaborn-swarmplot-2.png

Draw horizontal swarms:

>>> ax = sns.swarmplot(x="total_bill", y="day", data=tips)

http://seaborn.pydata.org/_images/seaborn-swarmplot-3.png

Color the points using a second categorical variable:

>>> ax = sns.swarmplot(x="day", y="total_bill", hue="sex", data=tips)

http://seaborn.pydata.org/_images/seaborn-swarmplot-4.png

Split each level of the hue variable along the categorical axis:

>>> ax = sns.swarmplot(x="day", y="total_bill", hue="smoker",
...                    data=tips, palette="Set2", dodge=True)

http://seaborn.pydata.org/_images/seaborn-swarmplot-5.png

Control swarm order by passing an explicit order:

>>> ax = sns.swarmplot(x="time", y="tip", data=tips,
...                    order=["Dinner", "Lunch"])

http://seaborn.pydata.org/_images/seaborn-swarmplot-6.png

Plot using larger points:

>>> ax = sns.swarmplot(x="time", y="tip", data=tips, size=6)

http://seaborn.pydata.org/_images/seaborn-swarmplot-7.png

Draw swarms of observations on top of a box plot:

>>> ax = sns.boxplot(x="tip", y="day", data=tips, whis=np.inf)
>>> ax = sns.swarmplot(x="tip", y="day", data=tips, color=".2")

http://seaborn.pydata.org/_images/seaborn-swarmplot-8.png

Draw swarms of observations on top of a violin plot:

>>> ax = sns.violinplot(x="day", y="total_bill", data=tips, inner=None)
>>> ax = sns.swarmplot(x="day", y="total_bill", data=tips,
...                    color="white", edgecolor="gray")

http://seaborn.pydata.org/_images/seaborn-swarmplot-9.png

Use catplot() to combine a swarmplot() and a FacetGrid. This allows grouping within additional categorical variables. Using catplot() is safer than using FacetGrid directly, as it ensures synchronization of variable order across facets:

>>> g = sns.catplot(x="sex", y="total_bill",
...                 hue="smoker", col="time",
...                 data=tips, kind="swarm",
...                 height=4, aspect=.7);

http://seaborn.pydata.org/_images/seaborn-swarmplot-10.png