Update plot outputs
This commit is contained in:
parent
0b5f8be42a
commit
339c27b99b
3 changed files with 10 additions and 7 deletions
|
@ -81,18 +81,21 @@ ax3d.quiver3D(
|
||||||
ax3d.set(xlabel="x (cm)", ylabel="y (cm)", zlabel="z (cm)")
|
ax3d.set(xlabel="x (cm)", ylabel="y (cm)", zlabel="z (cm)")
|
||||||
|
|
||||||
theta = np.angle(raw_ts["x"] + 1j * raw_ts["y"]).mean()
|
theta = np.angle(raw_ts["x"] + 1j * raw_ts["y"]).mean()
|
||||||
fig2dv, ax2dv = plt.subplots()
|
fig2dv, ax2dv = plt.subplots(figsize=(5/2.54, 2/3*10/2.54), dpi=200, constrained_layout=True)
|
||||||
x0 = ts_flt["x"] * np.cos(theta) + ts_flt["y"] * np.sin(theta)
|
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(
|
ax2dv.quiver(
|
||||||
x0[:-1],
|
x0[:-1],
|
||||||
z0[:-1],
|
z0[:-1],
|
||||||
np.diff(x0)[:],
|
np.diff(x0)[:],
|
||||||
np.diff(z0)[:],
|
np.diff(z0)[:],
|
||||||
color="#0066ff",
|
color="k",
|
||||||
|
scale_units="xy",
|
||||||
|
scale=1,
|
||||||
)
|
)
|
||||||
ax2dv.grid()
|
ax2dv.grid(c="k", alpha=.2)
|
||||||
ax2dv.set(aspect="equal")
|
ax2dv.set(aspect="equal", xlabel="x (cm)", ylabel="z (cm)")
|
||||||
fig2dv.savefig("out_orbitals.pdf")
|
fig2dv.savefig("out_orbitals.pdf")
|
||||||
|
fig2dv.savefig("out_orbitals.jpg")
|
||||||
|
|
||||||
plt.show()
|
plt.show()
|
||||||
|
|
|
@ -17,6 +17,6 @@ tf = 14300
|
||||||
x0 = -150
|
x0 = -150
|
||||||
|
|
||||||
[post]
|
[post]
|
||||||
out = out_post
|
out = out_post_refined
|
||||||
x = -50
|
x = -50
|
||||||
z = 5
|
z = 5
|
||||||
|
|
|
@ -87,4 +87,4 @@ with plt.rc_context(
|
||||||
loc="center left",
|
loc="center left",
|
||||||
)
|
)
|
||||||
|
|
||||||
plt.show()
|
fig.savefig(out.joinpath("fig.pdf"))
|
||||||
|
|
Loading…
Reference in a new issue