plot_against_benchmark.py

def plot_against_benchmark(series, benchmark, figsize = (13, 10), two_axis = False):
    """
    Plots a series against a benchmark.

    :param series: The series to be plotted.
    :param benchmark: The benchmark to be plotted.

    :return: The plot.
    """

    plt.figure(figsize = figsize)
    
    ax1 = series.plot(label=series.name, color='blue', legend = True)

    if two_axis:
        ax2 = ax1.twinx()
        ax2.plot(benchmark, label=benchmark.name, color='red')
    else:
        benchmark.plot(label=benchmark.name, color='red', ax=ax1, legend = True)

    return plt.gcf()