From ea4979bfc1b8a0521f680bad613062a14b36eb8f Mon Sep 17 00:00:00 2001
From: Jaslo Ziska <jaslo.ziska@fu-berlin.de>
Date: Mon, 17 Jul 2023 11:37:35 +0200
Subject: [PATCH] Update sheet11.py

---
 Jaslo/sheet11.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/Jaslo/sheet11.py b/Jaslo/sheet11.py
index 53f5d93..17db462 100644
--- a/Jaslo/sheet11.py
+++ b/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()
-- 
GitLab