more accessible plotsé
This commit is contained in:
@@ -45,14 +45,14 @@ def display_bkg(data, background, std_bkg, headers, histograms=None, binning=Non
|
||||
date_time = np.array([datetime.strptime(d, "%Y-%m-%d %H:%M:%S") for d in date_time])
|
||||
date_err = np.array([timedelta(seconds=headers[i]["exptime"] / 2.0) for i in range(len(headers))])
|
||||
filt = np.array([headers[i]["filtnam1"] for i in range(len(headers))])
|
||||
dict_filt = {"POL0": "r", "POL60": "g", "POL120": "b"}
|
||||
c_filt = np.array([dict_filt[f] for f in filt])
|
||||
dict_filt = {"POL0": "yo", "POL60": "bv", "POL120": "rs"}
|
||||
c_filt = np.array([dict_filt[f][0] for f in filt])
|
||||
|
||||
fig, ax = plt.subplots(figsize=(10, 6), constrained_layout=True)
|
||||
fig, ax = plt.subplots(figsize=(10, 4), constrained_layout=True)
|
||||
ax.errorbar(date_time, background * convert_flux, xerr=date_err, yerr=std_bkg * convert_flux, fmt="+k", markersize=0, ecolor=c_filt)
|
||||
for f in np.unique(filt):
|
||||
mask = [fil == f for fil in filt]
|
||||
ax.scatter(date_time[mask], background[mask] * convert_flux[mask], color=dict_filt[f], label="{0:s}".format(f))
|
||||
ax.errorbar(date_time, background * convert_flux, xerr=date_err, yerr=std_bkg * convert_flux, fmt="+k", markersize=0, ecolor=c_filt)
|
||||
ax.scatter(date_time[mask], background[mask] * convert_flux[mask], color=dict_filt[f][0], marker=dict_filt[f][1], label="{0:s}".format(f))
|
||||
# Date handling
|
||||
locator = mdates.AutoDateLocator()
|
||||
formatter = mdates.ConciseDateFormatter(locator)
|
||||
@@ -69,7 +69,7 @@ def display_bkg(data, background, std_bkg, headers, histograms=None, binning=Non
|
||||
this_savename += "_background_flux.pdf"
|
||||
else:
|
||||
this_savename = savename[:-4] + "_background_flux" + savename[-4:]
|
||||
fig.savefig(path_join(plots_folder, this_savename), bbox_inches="tight")
|
||||
fig.savefig(path_join(plots_folder, this_savename), bbox_inches="tight", facecolor="None", edgecolor="None")
|
||||
|
||||
if histograms is not None:
|
||||
filt_obs = {"POL0": 0, "POL60": 0, "POL120": 0}
|
||||
@@ -111,7 +111,7 @@ def display_bkg(data, background, std_bkg, headers, histograms=None, binning=Non
|
||||
this_savename += "_histograms.pdf"
|
||||
else:
|
||||
this_savename = savename[:-4] + "_histograms" + savename[-4:]
|
||||
fig_h.savefig(path_join(plots_folder, this_savename), bbox_inches="tight")
|
||||
fig_h.savefig(path_join(plots_folder, this_savename), bbox_inches="tight", facecolor="None", edgecolor="None")
|
||||
|
||||
fig2, ax2 = plt.subplots(figsize=(10, 10))
|
||||
data0 = data[0] * convert_flux[0]
|
||||
@@ -144,7 +144,7 @@ def display_bkg(data, background, std_bkg, headers, histograms=None, binning=Non
|
||||
this_savename += "_" + filt + "_background_location.pdf"
|
||||
else:
|
||||
this_savename = savename[:-4] + "_" + filt + "_background_location" + savename[-4:]
|
||||
fig2.savefig(path_join(plots_folder, this_savename), bbox_inches="tight")
|
||||
fig2.savefig(path_join(plots_folder, this_savename), bbox_inches="tight", facecolor="None", edgecolor="None")
|
||||
if rectangle is not None:
|
||||
plot_obs(
|
||||
data,
|
||||
|
||||
@@ -2913,7 +2913,7 @@ class pol_map(object):
|
||||
b_tf = Button(ax_tf, r"$F_{\lambda}$")
|
||||
b_pf = Button(ax_pf, r"$F_{\lambda} \cdot P$")
|
||||
b_p = Button(ax_p, r"$P$")
|
||||
b_pa = Button(ax_pa, r"$\theta_{P}$")
|
||||
b_pa = Button(ax_pa, r"$\Psi$")
|
||||
b_snri = Button(ax_snri, r"$I / \sigma_{I}$")
|
||||
b_snrp = Button(ax_snrp, r"$P / \sigma_{P}$")
|
||||
|
||||
@@ -3220,8 +3220,8 @@ class pol_map(object):
|
||||
ax.set_ylim(0, self.data.shape[0])
|
||||
plt.rcParams.update({"font.size": 14})
|
||||
fig.colorbar(im, ax=ax, aspect=50, shrink=0.75, pad=0.025, label=label)
|
||||
if format_cursor_data is not None:
|
||||
self.im.format_cursor_data = format_cursor_data
|
||||
# if format_cursor_data is not None:
|
||||
# self.im.format_cursor_data = format_cursor_data
|
||||
plt.rcParams.update({"font.size": 10})
|
||||
fig.canvas.draw_idle()
|
||||
return im
|
||||
|
||||
Reference in New Issue
Block a user