black (still without mortimer)

This commit is contained in:
Noe Brucy
2023-01-30 12:12:14 +01:00
parent 8c3db9b7cb
commit 1a8db42d29
11 changed files with 474 additions and 99 deletions
+5 -5
View File
@@ -320,14 +320,14 @@ class InteractiveGUI:
10 ** (a * centers + b),
"--",
color="navy",
label=r"a = {:.3g}, $R^2$ = {:.3g}".format(a, r ** 2),
label=r"a = {:.3g}, $R^2$ = {:.3g}".format(a, r**2),
)
plt.yscale("log")
plt.title("PDF")
else:
self.step.set_ydata(values)
self.fit.set_ydata(10 ** (a * centers + b))
self.fit.set_label(r"a = {:.3g}, $R^2$ = {:.3g}".format(a, r ** 2))
self.fit.set_label(r"a = {:.3g}, $R^2$ = {:.3g}".format(a, r**2))
plt.legend()
# PROFILE
@@ -382,7 +382,7 @@ class InteractiveGUI:
a * np.log10(x[mask_fit_prof]) + b,
"--",
color="navy",
label=r"a = {:.3g}, $R^2$ = {:.3g}".format(a, r ** 2),
label=r"a = {:.3g}, $R^2$ = {:.3g}".format(a, r**2),
)
self.ax_profile.set_xlabel(r"$r$")
self.ax_profile.set_ylabel(r"$\log(\rho)$")
@@ -390,7 +390,7 @@ class InteractiveGUI:
self.prof.set_data(x, rho_prof)
self.prof_z.set_data(z_vert, rho_vert)
self.fit_prof.set_data(x[mask_fit_prof], a * np.log10(x[mask_fit_prof]) + b)
self.fit_prof.set_label(r"a = {:.3g}, $R^2$ = {:.3g}".format(a, r ** 2))
self.fit_prof.set_label(r"a = {:.3g}, $R^2$ = {:.3g}".format(a, r**2))
plt.legend()
plt.draw()
@@ -486,7 +486,7 @@ class InteractiveGUI:
y = np.linspace(self.im_extent[2], self.im_extent[3], self.shape[1])
self.xx, self.yy = np.meshgrid(x, y)
self.xy = np.column_stack((self.xx.flatten(), self.yy.flatten()))
self.rr = np.sqrt(self.xx ** 2 + self.yy ** 2)
self.rr = np.sqrt(self.xx**2 + self.yy**2)
self.rhocs = np.column_stack((self.frho_map.flatten(), self.fcs_map.flatten()))
self.rmin = self.pp.params.disk.rmin_pdf / 2.0