lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-03-15 14:50:25 +01:00
Code Issues Projects Releases Packages Wiki Activity

feat: argument hide_sigma added to Corr.show which allows to

Browse source 
automatically hide data points with large errors.
This commit is contained in:
Fabian Joswig 2022-05-02 10:12:23 +01:00
parent 9e3e84ffcd
commit d0f96f2bf8
1 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
pyerrors/correlators.py
View file

@ -703,7 +703,7 @@ class Corr:
self.prange = prange
return
def show(self, x_range=None, comp=None, y_range=None, logscale=False, plateau=None, fit_res=None, ylabel=None, save=None, auto_gamma=False):
def show(self, x_range=None, comp=None, y_range=None, logscale=False, plateau=None, fit_res=None, ylabel=None, save=None, auto_gamma=False, hide_sigma=None):
"""Plots the correlator using the tag of the correlator as label if available.
Parameters
@ -725,6 +725,8 @@ class Corr:
path to file in which the figure should be saved
auto_gamma : bool
Apply the gamma method with standard parameters to all correlators and plateau values before plotting.
hide_sigma : float
Hides data points from the first value on which is consistent with zero within 'hide_sigma' standard errors.
"""
if self.N != 1:
raise Exception("Correlator must be projected before plotting")
@ -739,7 +741,11 @@ class Corr:
ax1 = fig.add_subplot(111)
x, y, y_err = self.plottable()
ax1.errorbar(x, y, y_err, label=self.tag)
if hide_sigma:
hide_from = np.argmax((hide_sigma * np.array(y_err)) > np.abs(y)) - 1
else:
hide_from = None
ax1.errorbar(x[:hide_from], y[:hide_from], y_err[:hide_from], label=self.tag)
if logscale:
ax1.set_yscale('log')
else:
@ -758,7 +764,11 @@ class Corr:
if auto_gamma:
corr.gamma_method()
x, y, y_err = corr.plottable()
plt.errorbar(x, y, y_err, label=corr.tag, mfc=plt.rcParams['axes.facecolor'])
if hide_sigma:
hide_from = np.argmax((hide_sigma * np.array(y_err)) > np.abs(y)) - 1
else:
hide_from = None
plt.errorbar(x[:hide_from], y[:hide_from], y_err[:hide_from], label=corr.tag, mfc=plt.rcParams['axes.facecolor'])
else:
raise Exception("'comp' must be a correlator or a list of correlators.")