Merge pull request #88 from Team-RADDISH/tk/plot-std

Fix x-axis in ESS plot to start from 1
Team-RADDISH · Jun 4, 2020 · 21514d3 · 21514d3
2 parents ba7a28d + 081f59f
commit 21514d3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/extra/Plot_tdac_output.ipynb b/extra/Plot_tdac_output.ipynb
@@ -161,7 +161,8 @@
     "for ts in list(fh[\"weights\"])[1:]:\n",
     "    ess.append(1/sum(fh[\"weights\"][ts][:]**2))\n",
     "fig = plt.figure()\n",
-    "plt.plot(ess[1:])\n",
+    "t = np.arange(1,np.size(ess))\n",
+    "plt.plot(t,ess[1:])\n",
     "plt.xlabel('Time step')\n",
     "plt.ylabel('Estimated Sample Size (1 / sum(weight^2))');"
    ]