plotlib Namespace Reference

Functions
def	histplot (data_x, data_y, error=None, label=None, **kwargs)

Function Documentation

histplot()

def plotlib.histplot	(		data_x,
			data_y,
			error = None,
			label = None,
		**	kwargs
	)

Definition at line 18 of file plotlib.py.

def histplot(data_x, data_y, error=None, label=None, **kwargs):
    settings = {
        "xlabel" : r"$p_T\:[GeV]$",
        "ylabel": 'Efficiency',
        "title1": r"ATLAS",
        "title1_1": r"Simulation",
        "title2": r"$\sqrt{s}=13\:TeV$",
        "filename": "trunontest",
        "log_y":False,
        "norm":False,
        "scale":1,
        "upper_y": 1.7,
        "errorbar_limit": None,
        }
    for each_key in kwargs.items():
        settings[each_key[0]] = kwargs[each_key[0]]
    plt.figure(figsize=(8, 6))
    remove_label = False
    if label is None:
        remove_label = True
        label = ["a"] * len(data_x)
    for i in range(len(data_x)):
        if error is not None:
            if settings["errorbar_limit"] is not None:
                allerror = []
                for each_error, each_yi in zip(error[i], data_y[i]):
                    uperror = each_error
                    if each_yi + each_error > settings["errorbar_limit"]:
                        uperror = settings["errorbar_limit"] - each_yi
                    allerror.append([each_error, uperror])
            else:
                allerror = error[i]
            style = '.-'
            if len(data_x) == 1:
                style = 'k.-'
            plt.errorbar(data_x[i], data_y[i], yerr=np.array(allerror).T, label=label[i], fmt=style)
        else:
            plt.plot(data_x[i], data_y[i], label=label[i])
    ax1 = plt.gca()
 
    plt.legend(loc='upper right', prop={'size': 25}, frameon=False)
    if remove_label:
        ax1.get_legend().remove()
 
    ax1.set_ylim([0, max([y for x in data_y for y in x])* settings["upper_y"]])
    ax1.text(0.05, 1.55 / 1.7, settings['title1'], fontsize=25, transform=ax1.transAxes, style='italic', fontweight='bold')
    ax1.text(0.28, 1.55/ 1.7, settings['title1_1'], fontsize=25, transform=ax1.transAxes)
    ax1.text(0.05, 1.40 / 1.7, settings['title2'], fontsize=23, transform=ax1.transAxes)
 
    plt.tick_params(labelsize=16)
    plt.tick_params(labelsize=16)
    plt.ylabel(settings['ylabel'], fontsize=20)
    plt.xlabel(settings['xlabel'], fontsize=20)
    plt.savefig(settings['filename'] + '.pdf', bbox_inches='tight')
    plt.close()