plot a grouped barplot

grouped_barplot.py

import matplotlib.pyplot as plt
import numpy as np

from typing import List, Optional


def grouped_barplot(
    data,
    clabels: List[str],
    xlabels: List[str],
    gap: float = 0.3,
    show_legend: bool = True,
    show_bar_labels: bool = True,
    ax: Optional[plt.Axes] = None,
):
    
    if ax is None:
        _, ax = plt.subplots()
    
    x = np.arange(len(xlabels))  # the label locations
    width = (1 - gap)/len(clabels)  # the width of the bars

    for i, (cdata, clabel) in enumerate(zip(data, clabels)):
        rects = ax.bar(x - ((1 - gap) / 2) + i * width, cdata, width, label=clabel)
        if show_bar_labels:
            ax.bar_label(rects, padding=3)

    # Add some text for labels, title and custom x-axis tick labels, etc.
    ax.set_xticks(x, xlabels)
    if show_legend:
        ax.legend()
    
    return ax