#! /usr/bin/python3 # ================================================================== # Python code for 2D random walk # # From: www.geeksforgeeks.org/random-walk-implementation-python/ # ================================================================== import numpy import pylab import random # defining the number of steps n = 100 #creating two array for containing x and y coordinate #of size equals to the number of size and filled up with 0's x = numpy.zeros(n) y = numpy.zeros(n) # filling the coordinates with random variables for i in range(1, n): val = random.randint(1, 4) if val == 1: x[i] = x[i - 1] + 1 y[i] = y[i - 1] elif val == 2: x[i] = x[i - 1] - 1 y[i] = y[i - 1] elif val == 3: x[i] = x[i - 1] y[i] = y[i - 1] + 1 else: x[i] = x[i - 1] y[i] = y[i - 1] - 1 # plot random walk information pylab.title("Random Walk ($n = " + str(n) + "$ steps)") pylab.plot(x, y) pylab.savefig("rand_walk"+str(n)+".png",bbox_inches="tight",dpi=600) pylab.show()