Update sheet11.py

Jaslo/sheet11.py

@@ -118,7 +118,10 @@ msd = np.zeros(len(rp))
 for i in tqdm(range(1, len(msd))):
     msd[i] = np.mean(np.sum((rp[i:] - rp[:-i])**2, axis=2))
 
-plt.loglog(t[n_eq:], msd)
+plt.loglog(t[n_eq:] - t_eq, msd)
+plt.xlabel(r"$\log t$")
+plt.ylabel(r"$\log \delta r^2(t)$")
+plt.savefig("problem11.1a.png", dpi=300)
 plt.show()
 
 # b)
@@ -166,6 +169,14 @@ msdv = np.zeros(len(rv))
 for i in tqdm(range(1, len(msdv))):
     msdv[i] = np.mean(np.sum((rv[i:] - rv[:-i])**2, axis=2))
 
-plt.loglog(t[n_eq:], msd / t[n_eq:])
-plt.loglog(t[n_eq:], msdv / t[n_eq:])
+plt.loglog(t[n_eq:], msd)
+plt.loglog(t[n_eq:], msdv)
+plt.show()
+
+plt.loglog(t[n_eq:] - t_eq, msd / t[n_eq:], label="Langevin")
+plt.loglog(t[n_eq:] - t_eq, msdv / t[n_eq:], label="Verlet")
+plt.legend()
+plt.xlabel(r"$\log t$")
+plt.ylabel(r"$\log(\delta r^2(t) / t)$")
+plt.savefig("problem11.1b.png", dpi=300)
 plt.show()