data/processing/orbitals.py

@@ -81,21 +81,18 @@ ax3d.quiver3D(
 ax3d.set(xlabel="x (cm)", ylabel="y (cm)", zlabel="z (cm)")
 
 theta = np.angle(raw_ts["x"] + 1j * raw_ts["y"]).mean()
-fig2dv, ax2dv = plt.subplots(figsize=(5/2.54, 2/3*10/2.54), dpi=200, constrained_layout=True)
+fig2dv, ax2dv = plt.subplots()
 x0 = ts_flt["x"] * np.cos(theta) + ts_flt["y"] * np.sin(theta)
-#ax2dv.plot(x0, z0, c="#0066ff", lw=1)
+ax2dv.plot(x0, z0, c="#0066ff", lw=1)
 ax2dv.quiver(
     x0[:-1],
     z0[:-1],
     np.diff(x0)[:],
     np.diff(z0)[:],
-    color="k",
-    scale_units="xy",
-    scale=1,
+    color="#0066ff",
 )
-ax2dv.grid(c="k", alpha=.2)
-ax2dv.set(aspect="equal", xlabel="x (cm)", ylabel="z (cm)")
+ax2dv.grid()
+ax2dv.set(aspect="equal")
 fig2dv.savefig("out_orbitals.pdf")
-fig2dv.savefig("out_orbitals.jpg")
 
 plt.show()