bar.py

import matplotlib.pyplot as plt
from save import save_fig

def plot_bar(
    data: dict,
    save_path=None,
    figsize=(4, 3),
    xlabel="",
    ylabel="",
    rotate_xticks=45,
    y_lim=1.1,
    show_values=True,
    sort=False,
    color=None,
):
    """
    Plot a generic categorical bar chart.

    Args:
        data (dict): {category_name: value}
        save_path (str or None): path to save figure (pdf)
        figsize (tuple): figure size in inches
        xlabel (str)
        ylabel (str)
        rotate_xticks (int)
        y_lim (float): y-axis limit multiplier
        show_values (bool)
        sort (bool)
        color (str or None): bar color
    """

    if sort:
        items = sorted(data.items(), key=lambda x: x[1], reverse=True)
    else:
        items = data.items()

    categories, values = zip(*items)

    fig, ax = plt.subplots(figsize=figsize)

    bars = ax.bar(categories, values, color=color, edgecolor='none', linewidth=0)

    ax.spines["top"].set_visible(False)
    ax.spines["right"].set_visible(False)

    ax.set_xlabel(xlabel)
    ax.set_ylabel(ylabel)

    # 补丁级别，添加适应指定上下限
    # ax.set_ylim(0, max(values) * y_lim)
    if isinstance(y_lim, (tuple, list)):
        ax.set_ylim(y_lim)
    else:
        ax.set_ylim(0, max(values) * y_lim)

    if rotate_xticks:
        ax.set_xticklabels(categories, rotation=rotate_xticks, ha="right")

    if show_values:
        ax.bar_label(bars, padding=2, fontsize=8)

    plt.tight_layout()

    if save_path is not None:
        save_fig(save_path)

    plt.close()