Venn Plot

This module provides functionality for creating Venn and Euler diagrams from pandas DataFrames.

It is designed to visualize relationships between sets, highlighting intersections and differences between them.

Core Features

• Supports 2-set and 3-set Diagrams: Allows visualization of up to three overlapping sets.
• Venn and Euler Diagrams: Uses Venn diagrams by default; switches to Euler diagrams when vary_size=True.
• Customizable Colors and Labels: Automatically assigns colors and labels for subset representation.
• Dynamic Sizing: Adjusts circle sizes for Euler diagrams to reflect proportions.
• Title and Source Attribution: Optionally adds a title and source text.

Use Cases

• Set Comparisons: Identify shared and unique elements across two or three sets.
• Proportional Representation: Euler diagrams ensure area-accurate representation.
• Data Overlap Visualization: Helps in understanding relationships within categorical data.

Limitations and Warnings

• Only Supports 2 or 3 Sets: Does not extend to Venn diagrams with more than three sets.
• Pre-Aggregated Data Required: The module does not perform data aggregation; input data should already be structured correctly.

plot(df, labels, title=None, source_text=None, vary_size=False, figsize=None, ax=None, subset_label_formatter=None, **kwargs)

Plots a Venn or Euler diagram using subset sizes extracted from a DataFrame.

Parameters:

Name	Type	Description	Default
df	DataFrame	DataFrame with 'groups' and 'percent' columns.	required
labels	list[str]	Labels for the sets in the diagram.	required
title	str	Title of the plot. Defaults to None.	None
source_text	str	Source text for attribution. Defaults to None.	None
vary_size	bool	Whether to vary circle size based on subset sizes. Defaults to False.	False
figsize	tuple[int, int]	Size of the plot. Defaults to None.	None
ax	Axes	Matplotlib axes object to plot on. Defaults to None.	None
subset_label_formatter	callable	Function to format subset labels. Defaults to None.	None
**kwargs	dict[str, any]	Additional keyword arguments.	{}

Returns:

Name	Type	Description
SubplotBase	SubplotBase	The matplotlib axes object with the plotted diagram.

Raises:

Type	Description
ValueError	If the number of sets is not 2 or 3.

Source code in pyretailscience/plots/venn.py

def plot(
    df: pd.DataFrame,
    labels: list[str],
    title: str | None = None,
    source_text: str | None = None,
    vary_size: bool = False,
    figsize: tuple[int, int] | None = None,
    ax: Axes | None = None,
    subset_label_formatter: Callable | None = None,
    **kwargs: dict[str, any],
) -> SubplotBase:
    """Plots a Venn or Euler diagram using subset sizes extracted from a DataFrame.

    Args:
        df (pd.DataFrame): DataFrame with 'groups' and 'percent' columns.
        labels (list[str]): Labels for the sets in the diagram.
        title (str, optional): Title of the plot. Defaults to None.
        source_text (str, optional): Source text for attribution. Defaults to None.
        vary_size (bool, optional): Whether to vary circle size based on subset sizes. Defaults to False.
        figsize (tuple[int, int], optional): Size of the plot. Defaults to None.
        ax (Axes, optional): Matplotlib axes object to plot on. Defaults to None.
        subset_label_formatter (callable, optional): Function to format subset labels. Defaults to None.
        **kwargs: Additional keyword arguments.

    Returns:
        SubplotBase: The matplotlib axes object with the plotted diagram.

    Raises:
        ValueError: If the number of sets is not 2 or 3.
    """
    num_sets = len(labels)
    if num_sets not in {MIN_SUPPORTED_SETS, MAX_SUPPORTED_SETS}:
        raise ValueError("Only 2-set or 3-set Venn diagrams are supported.")

    colors = [COLORS["green"][500], COLORS["green"][800]]
    if num_sets == MAX_SUPPORTED_SETS:
        colors.append(COLORS["green"][200])

    zero_group = (0, 0) if num_sets == MIN_SUPPORTED_SETS else (0, 0, 0)
    percent_s = df.loc[df["groups"] != zero_group, ["groups", "percent"]].set_index("groups")["percent"]
    subset_sizes = percent_s.to_dict()

    subset_labels = {
        k: subset_label_formatter(v) if subset_label_formatter else str(v) for k, v in subset_sizes.items()
    }

    if ax is None:
        _, ax = plt.subplots(figsize=figsize)

    diagram_class = EulerDiagram if vary_size else VennDiagram
    diagram = diagram_class(
        set_labels=labels,
        subset_sizes=subset_sizes,
        subset_labels=subset_labels,
        set_colors=colors,
        ax=ax,
        **kwargs,
    )

    center_x, center_y, displacement = 0.5, 0.5, 0.1
    for text in diagram.set_label_artists:
        text.set_fontproperties(GraphStyles.POPPINS_REG)
        text.set_fontsize(GraphStyles.DEFAULT_AXIS_LABEL_FONT_SIZE)
        if num_sets == MAX_SUPPORTED_SETS and not vary_size:
            x, y = text.get_position()
            direction_x, direction_y = x - center_x, y - center_y
            scale = displacement / (direction_x**2 + direction_y**2) ** 0.5
            text.set_position((x + scale * direction_x, y + scale * direction_y))

    for subset_id in subset_sizes:
        if subset_id not in diagram.subset_label_artists:
            continue
        text = diagram.subset_label_artists[subset_id]
        text.set_fontproperties(GraphStyles.POPPINS_REG)

    if title:
        ax.set_title(
            title,
            fontproperties=GraphStyles.POPPINS_SEMI_BOLD,
            fontsize=GraphStyles.DEFAULT_TITLE_FONT_SIZE,
            pad=GraphStyles.DEFAULT_TITLE_PAD + 20,
        )

    if source_text is not None:
        ax.set_xticklabels([], visible=False)
        gu.add_source_text(ax=ax, source_text=source_text, is_venn_diagram=True)

    return ax