plothist 1.3.2__py3-none-any.whl → 1.5.0__py3-none-any.whl

plothist/examples/2d_hist/2d_hist_simple_discrete_colormap.py

@@ -0,0 +1,42 @@
+"""
+2D histogram, discrete colormap
+===============================
+
+Plot a 2D histogram with ``plot_2d_hist()`` with a discrete colormap.
+"""
+
+from plothist_utils import get_dummy_data
+
+df = get_dummy_data()
+
+from plothist import make_2d_hist, plot_2d_hist
+
+name_x = "variable_0"
+name_y = "variable_1"
+
+# We only take a subset of the data
+nentries = 1000
+
+h = make_2d_hist([df[name_x][:nentries], df[name_y][:nentries]])
+
+###
+from matplotlib.colors import ListedColormap
+
+from plothist import get_color_palette
+
+# 0 entries will be white, the rest will have one color from the plasma colormap per entry value
+cmap = ListedColormap(
+    ["white", *list(get_color_palette("plasma", int(h.values().max()) * 2 - 1))]
+)
+
+fig, ax, ax_colorbar = plot_2d_hist(
+    h, colorbar_kwargs={"label": "Entries"}, pcolormesh_kwargs={"cmap": cmap}
+)
+
+ax.set_xlabel(name_x)
+ax.set_ylabel(name_y)
+
+ax.set_xlim(-4.5, 4.5)
+ax.set_ylim(-4.5, 4.5)
+
+fig.savefig("2d_hist_simple_discrete_colormap.svg", bbox_inches="tight")

plothist/examples/2d_hist/2d_hist_uneven.py

@@ -0,0 +1,28 @@
+"""
+2D histogram, uneven binning
+============================
+
+Plot a 2D histogram with uneven binning.
+"""
+
+from plothist_utils import get_dummy_data
+
+df = get_dummy_data()
+
+###
+from plothist import make_2d_hist, plot_2d_hist
+
+name_x = "variable_0"
+name_y = "variable_1"
+# Bins [-10,0], [0,10] for variable 1,
+# and bins [-10,-5], [-5,0], [0,5], [5,10] for variable 2
+bins = [[-10, 0, 10], [-10, -5, 0, 5, 10]]
+
+h = make_2d_hist([df[name_x], df[name_y]], bins=bins)
+
+fig, ax, ax_colorbar = plot_2d_hist(h, colorbar_kwargs={"label": "Entries"})
+
+ax.set_xlabel(name_x)
+ax.set_ylabel(name_y)
+
+fig.savefig("2d_hist_uneven.svg", bbox_inches="tight")

plothist/examples/2d_hist/2d_hist_with_projections.py

@@ -0,0 +1,36 @@
+"""
+2D histogram with projections
+==============================
+
+Plot a 2D histogram with the two 1D projections.
+"""
+
+from plothist_utils import get_dummy_data
+
+df = get_dummy_data()
+
+###
+from plothist import make_2d_hist, plot_2d_hist_with_projections
+
+name_x = "variable_0"
+name_y = "variable_1"
+
+h = make_2d_hist([df[name_x], df[name_y]])
+
+(
+    fig,
+    ax_2d,
+    ax_x_projection,
+    ax_y_projection,
+    ax_colorbar,
+) = plot_2d_hist_with_projections(
+    h,
+    xlabel="variable_0",
+    ylabel="variable_1",
+    ylabel_x_projection="Entries",
+    xlabel_y_projection="Entries",
+    offset_x_labels=False,
+    colorbar_kwargs={"label": "Entries"},
+)
+
+fig.savefig("2d_hist_with_projections.svg", bbox_inches="tight")

plothist/examples/2d_hist/README.rst

@@ -0,0 +1,4 @@
+Plot 2D histograms
+------------------
+
+Gallery of images that are used in the section about plotting 2D histograms.

plothist/examples/README.rst

@@ -0,0 +1,7 @@
+Example gallery
+===============
+
+Gallery of images that are used in the doc.
+
+.. note::
+   Click on an image to see the source code that generates it.

plothist/examples/advanced/1d_comparison_advanced.py

@@ -0,0 +1,87 @@
+"""
+Multiple comparisons
+====================
+
+Compare two 1D histograms using the pull and ratio methods on the same plot.
+"""
+
+from plothist_utils import get_dummy_data
+
+df = get_dummy_data()
+
+###
+from plothist import (
+    create_comparison_figure,
+    get_color_palette,
+    make_hist,
+    plot_comparison,
+    plot_error_hist,
+    plot_hist,
+)
+
+name = "variable_1"
+category = "category"
+
+x1 = df[name][df[category] == 1]
+x2 = df[name][df[category] == 4]
+x3 = df[name][df[category] == 3]
+x4 = df[name][df[category] == 5]
+
+x_range = (-9, 9)
+
+h1 = make_hist(x3, bins=50, range=x_range)
+h2 = make_hist(x4, bins=50, range=x_range)
+h3 = make_hist(x1, bins=50, range=x_range)
+h4 = make_hist(x2, bins=50, range=x_range)
+
+# Create the 3 axes that we need for this plot
+fig, axes = create_comparison_figure(
+    figsize=(6, 6), nrows=3, gridspec_kw={"height_ratios": [5, 1, 1]}
+)
+
+# Assign each axes: 1 to plot the histograms and 2 for the comparison plots
+ax_main, ax1_comparison, ax2_comparison = axes
+
+# Get the red and the blue from the default color cycle
+colors = get_color_palette("ggplot", 2)
+
+# Here, we use step as a histtype to only draw the line
+plot_hist(h1, label="Train A", ax=ax_main, histtype="step", linewidth=1.2, density=True)
+plot_hist(h3, label="Train B", ax=ax_main, histtype="step", linewidth=1.2, density=True)
+# And then, to make the plot easier to read, we redraw them with stepfilled, which add color below the line
+plot_hist(
+    h1, ax=ax_main, histtype="stepfilled", color=colors[0], alpha=0.2, density=True
+)
+plot_hist(
+    h3, ax=ax_main, histtype="stepfilled", color=colors[1], alpha=0.2, density=True
+)
+
+# We plot 2 additional histograms with point style
+plot_error_hist(h2, label="Test A", ax=ax_main, color="blue", density=True)
+plot_error_hist(h4, label="Test B", ax=ax_main, color="red", density=True)
+
+# First comparison is using pulls. We also change the color of the bars to make the plot easier to read
+plot_comparison(
+    h2, h1, ax=ax1_comparison, comparison="pull", color=colors[0], alpha=0.7
+)
+# Second comparison is using the default "ratio". Same strategy as pulls
+plot_comparison(h4, h3, ax=ax2_comparison, color=colors[1], alpha=0.7)
+
+# Harmonize the range of each axes
+ax_main.set_xlim(x_range)
+ax1_comparison.set_xlim(x_range)
+ax2_comparison.set_xlim(x_range)
+
+# Set the labels for the different axes
+ax_main.set_ylabel("Entry density")
+ax1_comparison.set_ylabel("$Pull_{A}$")
+ax2_comparison.set_ylabel("$Ratio_{B}$")
+ax2_comparison.set_xlabel("Variable [unit]")
+
+# Add the legend
+ax_main.legend(loc="upper left")
+
+# Align the ylabels
+fig.align_ylabels()
+
+fig.savefig("1d_comparison_advanced.svg", bbox_inches="tight")

plothist/examples/advanced/1d_side_by_side_with_numbers.py

@@ -0,0 +1,81 @@
+"""
+Side-by-side categories, with numbers on top
+============================================
+
+Plot multiple 1D histograms with categories side by side, and add the number of entries on top of each bar.
+"""
+
+###
+import boost_histogram as bh
+import matplotlib.pyplot as plt
+import numpy as np
+
+from plothist import plot_hist
+
+rng = np.random.default_rng(83113111)
+
+# Integer categories
+categories = [-137, 12, 1234]
+axis = bh.axis.IntCategory(categories=categories)
+
+# Generate data for 3 histograms
+data = [
+    rng.choice(categories, 50),
+    rng.choice(categories, 30),
+    rng.choice(categories, 35),
+    rng.choice(categories, 30),
+]
+
+# Create and fill the histograms
+histos = [bh.Histogram(axis, storage=bh.storage.Weight()) for _ in range(len(data))]
+histos = [histo.fill(data[i]) for i, histo in enumerate(histos)]
+
+labels = [f"$h_{{{i}}}$" for i in range(len(histos))]
+colors = ["#348ABD", "#E24A33", "#988ED5", "#FBC15E"]
+
+# Plot the histogram
+fig, ax = plt.subplots()
+
+# Use a specificity of matplotlib: when a list of histograms is given, it will plot them side by side unless stacked=True or histtype is a "step" type.
+plot_hist(histos, ax=ax, label=labels, color=colors)
+
+
+# Add the number of entries on top of each bar
+# Get the correct shift in x-axis for each bar
+def calculate_shifts(width, n_bars):
+    half_width = width / 2
+    shift = np.linspace(-half_width, half_width, n_bars, endpoint=False)
+    shift += width / (2 * n_bars)
+    return shift
+
+
+bin_width = 0.8
+shift = calculate_shifts(bin_width, len(histos))
+
+# Loop over the histograms, add on top of each bar the number of entries
+for i, histo in enumerate(histos):
+    for j, value in enumerate(histo.values()):
+        ax.text(
+            j + 0.5 + shift[i],
+            value,
+            int(
+                value
+            ),  # If weighted, f"{height:.1f}" can be used as a better representation of the bin content
+            color="black",
+            ha="center",
+            va="bottom",
+        )
+
+# Set the x-ticks to the middle of the bins and label them
+ax.set_xlim(0, len(categories))
+ax.set_xticks([i + 0.5 for i in range(len(categories))])
+ax.set_xticklabels(categories)
+ax.minorticks_off()
+# Get nice looking y-axis ticks
+ax.set_ylim(top=int(np.max([np.max(histo.values()) for histo in histos]) * 1.5))
+
+ax.set_xlabel("Category")
+ax.set_ylabel("Entries")
+ax.legend()
+
+fig.savefig("1d_side_by_side_with_numbers.svg", bbox_inches="tight")

plothist/examples/advanced/README.rst

@@ -0,0 +1,4 @@
+Other advanced plots
+--------------------
+
+Gallery of images that are used in the other advanced plots section.

plothist/examples/advanced/asymmetry_comparison_advanced.py

@@ -0,0 +1,133 @@
+"""
+Advanced asymmetry comparison
+=============================
+
+Plot the asymmetry between 2 histograms and 2 functions.
+"""
+
+# Get dummy data
+from plothist_utils import get_dummy_data
+
+df = get_dummy_data()
+
+###
+from scipy.stats import norm
+
+from plothist import (
+    add_luminosity,
+    add_text,
+    create_comparison_figure,
+    make_hist,
+    plot_comparison,
+    plot_error_hist,
+    plot_function,
+)
+
+
+# Define some random functions that will be used as Data fit functions
+def f1(x):
+    return 4000 * norm.pdf(x, loc=-0.5, scale=1.6)
+
+
+def f2(x):
+    return 4000 * norm.pdf(x, loc=0.5, scale=1.6)
+
+
+name = "variable_1"
+category = "category"
+
+x1 = df[name][df[category] == 5]
+
+x_range = (-9, 9)
+
+# Create the histograms used as data
+h1 = make_hist(x1 - 2.5, bins=50, range=x_range)
+h2 = make_hist(x1 - 1.5, bins=50, range=x_range)
+
+# Create the figure
+fig, (ax_main, ax_comparison) = create_comparison_figure(
+    gridspec_kw={"height_ratios": [2, 1]}
+)
+
+# Define the marker style
+marker_1 = {
+    "color": "tab:red",
+    "markeredgecolor": "tab:red",
+    "ls": "None",
+    "fmt": "o",
+    "markersize": 5,
+    "label": "$Data_1$",
+}
+
+marker_2 = {
+    "color": "tab:blue",
+    "markerfacecolor": "white",
+    "markeredgecolor": "tab:blue",
+    "ls": "None",
+    "fmt": "o",
+    "markersize": 5,
+    "label": "$Data_2$",
+}
+
+# Plot the data
+plot_error_hist(
+    h1,
+    ax_main,
+    uncertainty_type="symmetrical",
+    density=False,
+    **marker_1,
+)
+
+plot_error_hist(
+    h2,
+    ax_main,
+    uncertainty_type="symmetrical",
+    density=False,
+    **marker_2,
+)
+
+# Plot the functions
+plot_function(f1, x_range, ax_main, color=marker_1["color"], label="Data$_1$ fit")
+plot_function(
+    f2, x_range, ax_main, color=marker_2["color"], linestyle="--", label="Data$_2$ fit"
+)
+
+# Plot the asymmetry comparison between the 2 histograms
+plot_comparison(
+    h1,
+    h2,
+    ax=ax_comparison,
+    h1_label=r"$Data_1$",
+    h2_label=r"$Data_2$",
+    comparison="asymmetry",
+    comparison_ylim=(-1, 1),
+)
+
+
+# Define the asymmetry of the 2 functions
+def asymmetry(x):
+    return (f1(x) - f2(x)) / (f1(x) + f2(x))
+
+
+# Plot the asymmetry of the 2 functions
+plot_function(asymmetry, x_range, ax_comparison, color="black")
+
+ax_main.legend()
+
+fig.align_ylabels()
+
+ax_main.set_xlim(x_range)
+ax_main.set_ylim(ymin=0)
+ax_main.set_ylabel("Entries")
+ax_main.legend()
+
+ax_comparison.set_xlim(x_range)
+ax_comparison.set_xlabel(name)
+
+add_text("Data and fit asymmetry comparison", ax=ax_comparison, x="right")
+
+add_luminosity(
+    collaboration="LMN 3", ax=ax_main, lumi="(1 + 0.3)", preliminary=True, x="left"
+)
+
+fig.savefig("asymmetry_comparison_advanced.svg", bbox_inches="tight")

plothist/examples/advanced/model_examples_flatten2D.py

@@ -0,0 +1,86 @@
+"""
+Flatten 2D histograms
+=====================
+
+This example shows how to flatten and display 2D histograms on one dimension.
+"""
+
+from plothist_utils import get_dummy_data
+
+df = get_dummy_data()
+
+###
+from plothist import (
+    flatten_2d_hist,
+    get_color_palette,
+    make_2d_hist,
+    plot_data_model_comparison,
+    plot_hist,
+)
+
+# Define the histograms
+
+key1 = "variable_1"
+key2 = "variable_2"
+# Bins [-12,0], [0,12] for variable 1,
+# and bins [-12,-5], [-5,0], [0,5], [5,12] for variable 2
+bins = [[-12, 0, 12], [-12, -5, 0, 5, 12]]
+category = "category"
+
+# Define datasets
+
+signal_mask = df[category] == 7
+data_mask = df[category] == 8
+
+background_categories = [0, 1, 2, 3, 4, 5, 6]
+background_categories_labels = [f"c{i}" for i in background_categories]
+background_categories_colors = get_color_palette(
+    "cubehelix", len(background_categories)
+)
+
+background_masks = [df[category] == p for p in background_categories]
+
+# Make histograms
+
+data_hist = make_2d_hist(
+    [df[key][data_mask] for key in [key1, key2]], bins=bins, weights=1
+)
+background_hists = [
+    make_2d_hist([df[key][mask] for key in [key1, key2]], bins=bins, weights=1)
+    for mask in background_masks
+]
+signal_hist = make_2d_hist(
+    [df[key][signal_mask] for key in [key1, key2]], bins=bins, weights=1
+)
+
+# Flatten the 2D histograms
+data_hist = flatten_2d_hist(data_hist)
+background_hists = [flatten_2d_hist(h) for h in background_hists]
+signal_hist = flatten_2d_hist(signal_hist)
+
+# Compare data and stacked histogram
+fig, ax_main, ax_comparison = plot_data_model_comparison(
+    data_hist=data_hist,
+    stacked_components=background_hists,
+    stacked_labels=background_categories_labels,
+    stacked_colors=background_categories_colors,
+    xlabel=rf"({key1} $\times$ {key2}) bin number",
+    ylabel="Entries",
+)
+
+plot_hist(
+    signal_hist,
+    ax=ax_main,
+    color="red",
+    label="Signal",
+    histtype="step",
+)
+
+for ax in [ax_main, ax_comparison]:
+    ax.set_xticks([i + 0.5 for i in range(8)])
+    ax.tick_params(axis="x", which="minor", bottom=False)
+ax_comparison.set_xticklabels([str(i + 1) for i in range(8)])
+
+ax_main.legend(ncol=3, fontsize=10, loc="upper left")
+
+fig.savefig("model_examples_flatten2D.svg", bbox_inches="tight")

plothist/examples/func_1d/README.rst

@@ -0,0 +1,4 @@
+Plot simple functions
+---------------------
+
+Gallery of images that are used in the basic function section.

plothist/examples/func_1d/fct_1d.py

@@ -0,0 +1,27 @@
+"""
+Function
+========
+
+Plot a 1D function with ``plot_function()``.
+"""
+
+###
+from matplotlib import pyplot as plt
+from scipy.stats import norm
+
+from plothist import plot_function
+
+
+# Define the gaussian function of mean=0.5 and std_dev=3
+def f(x):
+    return 1000 * norm.pdf(x, loc=0.5, scale=3)
+
+
+fig, ax = plt.subplots()
+
+plot_function(f, range=(-10, 10), ax=ax)
+
+ax.set_xlabel("x")
+ax.set_ylabel("f(x)")
+
+fig.savefig("fct_1d.svg", bbox_inches="tight")

plothist/examples/func_1d/fct_1d_stacked.py

@@ -0,0 +1,42 @@
+"""
+Stacked functions
+=================
+
+Plot stacked functions using ``plot_function()``.
+"""
+
+from scipy.stats import norm
+
+
+# Define the gaussian function of mean=0.5 and std_dev=3
+def f(x):
+    return 1000 * norm.pdf(x, loc=0.5, scale=3)
+
+
+###
+from matplotlib import pyplot as plt
+
+from plothist import plot_function
+
+
+# Another function
+def g(x):
+    return 1000 * norm.pdf(x, loc=2, scale=3)
+
+
+fig, ax = plt.subplots()
+
+plot_function(
+    [f, g],
+    range=(-10, 10),
+    ax=ax,
+    labels=["f1", "f2"],
+    stacked=True,
+)
+
+ax.set_xlabel("x")
+ax.set_ylabel("f(x)")
+ax.set_ylim(0)
+ax.legend()
+
+fig.savefig("fct_1d_stacked.svg", bbox_inches="tight")

plothist/examples/model_ex/README.rst

@@ -0,0 +1,4 @@
+Model and data
+--------------
+
+Gallery of images that are used in the plotting model and data section.

plothist/examples/model_ex/model_all_comparisons.py

@@ -0,0 +1,103 @@
+"""
+Data/model comparisons
+======================
+
+All supported comparisons between data and model.
+"""
+
+from plothist_utils import get_dummy_data
+
+df = get_dummy_data()
+
+from plothist import get_color_palette, make_hist
+
+# Define the histograms
+
+key = "variable_1"
+range = (-9, 12)
+category = "category"
+
+# Define masks
+signal_mask = df[category] == 7
+data_mask = df[category] == 8
+
+background_categories = [0, 1, 2]
+background_categories_labels = [f"c{i}" for i in background_categories]
+background_categories_colors = get_color_palette(
+    "cubehelix", len(background_categories)
+)
+
+background_masks = [df[category] == p for p in background_categories]
+
+# Make histograms
+data_hist = make_hist(df[key][data_mask], bins=50, range=range, weights=1)
+background_hists = [
+    make_hist(df[key][mask], bins=50, range=range, weights=1)
+    for mask in background_masks
+]
+
+# Optional: scale to data
+background_scaling_factor = data_hist.sum().value / sum(background_hists).sum().value
+background_hists = [background_scaling_factor * h for h in background_hists]
+
+###
+from plothist import (
+    add_text,
+    create_comparison_figure,
+    plot_comparison,
+    plot_data_model_comparison,
+    set_fitting_ylabel_fontsize,
+)
+
+fig, axes = create_comparison_figure(
+    figsize=(6, 13),
+    nrows=6,
+    gridspec_kw={"height_ratios": [3, 1, 1, 1, 1, 1]},
+    hspace=0.3,
+)
+background_sum = sum(background_hists)
+
+plot_data_model_comparison(
+    data_hist=data_hist,
+    stacked_components=background_hists,
+    stacked_labels=background_categories_labels,
+    stacked_colors=background_categories_colors,
+    xlabel="",
+    ylabel="Entries",
+    comparison="ratio",
+    fig=fig,
+    ax_main=axes[0],
+    ax_comparison=axes[1],
+)
+
+add_text(
+    r"Multiple data-model comparisons, $\mathbf{with}$ model uncertainty",
+    ax=axes[0],
+)
+add_text(r'  $\mathbf{→}$ comparison = "ratio"', ax=axes[1], fontsize=13)
+
+for k_comp, comparison in enumerate(
+    ["split_ratio", "pull", "relative_difference", "difference"], start=2
+):
+    ax_comparison = axes[k_comp]
+
+    plot_comparison(
+        data_hist,
+        background_sum,
+        ax=ax_comparison,
+        comparison=comparison,
+        xlabel="",
+        h1_label="Data",
+        h2_label="Pred.",
+        h1_uncertainty_type="asymmetrical",
+    )
+    add_text(
+        rf'  $\mathbf{{→}}$ comparison = "{comparison}"',
+        ax=ax_comparison,
+        fontsize=13,
+    )
+    set_fitting_ylabel_fontsize(ax_comparison)
+
+axes[-1].set_xlabel(key)
+
+fig.savefig("model_all_comparisons.svg", bbox_inches="tight")