Diffusion.py

From Werner KRAUTH

Revision as of 05:26, 6 June 2024 Werner (Talk | contribs) ← Previous diff		Revision as of 05:27, 6 June 2024 Werner (Talk | contribs) Next diff →
Line 1:		Line 1:
	import random		import random
	import matplotlib.pyplot as plt		import matplotlib.pyplot as plt
-		+
	N = 5		N = 5
	data = []		data = []

---

Revision as of 05:27, 6 June 2024

import random
import matplotlib.pyplot as plt

N = 5
data = []
tmax = 10
for stat in range(100000):
   pos = 0
   for t in range(tmax):
      pos = min(max(pos + random.choice([-1, 0, 1]), 0), N - 1)
   data.append(pos)
plt.title('diffusion starting at $x=0$,  $t = $' + str(tmax))
plt.hist(data, bins=N, range=(-0.5, N-0.5), density=True)
plt.savefig('diffusion.png')
plt.show()