/Python/1d random walk in python stack exchange

1d random walk in python stack exchange

# Python code for 1-D random walk. 
import random 
import numpy as np 
import matplotlib.pyplot as plt 

# Probability to move up or down 
prob = [0.05, 0.95]   

n = 1000 # number of steps

# statically defining the starting position 
start = 2  
positions = [start] 

# creating the random points 
rr = np.random.random(n) 
downp = rr < prob[0] 
upp = rr > prob[1] 

t = 1

step = (1/n)**0.5

for idownp, iupp in zip(downp, upp): 
    down = step if idownp and positions[-1] > 1 else 0
    up = step if iupp and positions[-1] < 4 else 0
    positions.append(positions[-1] - down + up) 

# plotting down the graph of the random walk in 1D 
x = [i*t/n for i in range(n+1)]
plt.plot(x, positions) 
plt.xlabel('Time (seconds)')
plt.ylabel('Distance')
plt.title(f"Random Walk ({n} steps in {t} seconds)")
plt.grid(True)
plt.savefig("random_walk.png")


plt.show()

Source: stackoverflow.com

Similar codes

2 votes150 views3 answers
Typeerror: 'list' object is not callable
Python
0 vote189 views1 answer
Missingno python
Python
6 votes282 views3 answers
How to split an input in python by comma
Python
0 vote261 views1 answer
Exception Type: AssertionError Exception Value: Expected a `Response`, `HttpResponse` or `HttpStreamingResponse` to be returned from the view, but received a `<class 'django.db.models.query.QuerySet'>`
Python Django
1 vote206 views3 answers
Check space in string python
Python
0 vote201 views1 answer
Sns prevent legend
Python
1 vote273 views1 answer
Django queryset average of unique values
Python Django
0 vote222 views1 answer
Tanimoto coefficient rdkit
Python
0 vote193 views2 answers
How to use django-filters with viewset
Python
4 votes197 views1 answer
Loca value and drop pandas dataframe
Python