-
Notifications
You must be signed in to change notification settings - Fork 0
/
randomwalk.py
65 lines (59 loc) · 1.82 KB
/
randomwalk.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt
def passo_rand(x_0, periodo):
coef_a = periodo//3
coef_b = periodo//7
return ((x_0*coef_a + coef_b)%periodo)
def caminhante(eixo, passo, posicao):
eixos = ['x', 'y', 'z']
for i in range(len(posicao)):
if eixos.index(eixo) == i:
posicao[i].append(posicao[i][-1]+passo)
else:
posicao[i].append(posicao[i][-1])
return posicao
passos = 1000
caminhadas = 10
periodo = 12345678559098877
val = 0
fig = plt.figure()
ax = fig.gca(projection='3d')
for j in range(caminhadas):
posicao = [[0],[0],[0]] #X, Y, Z
for i in range(passos):
print(posicao)
val = passo_rand(val, periodo)
if val <= (periodo/6):
eixo='x'
passo=+1
posicao=caminhante(eixo, passo, posicao)
print('1')
elif val>(periodo/6) and val<=(2*periodo/6):
eixo='x'
passo=-1
posicao=caminhante(eixo, passo, posicao)
print('2')
elif val>(2*periodo/6) and val<=(3*periodo/6):
eixo='y'
passo=+1
posicao=caminhante(eixo, passo, posicao)
print('3')
elif val>(3*periodo/6) and val<=(4*periodo/6):
eixo='y'
passo=-1
posicao=caminhante(eixo, passo, posicao)
print('4')
elif val>(4*periodo/6) and val<=(5*periodo/6):
eixo='z'
passo=+1
posicao=caminhante(eixo, passo, posicao)
print('5')
elif val>(5*periodo/6) and val<=(6*periodo/6):
eixo='z'
passo=-1
posicao=caminhante(eixo, passo, posicao)
print('6')
ax.plot(posicao[0], posicao[1], posicao[2], label=f'Caminhada {j+1}')
ax.legend()
plt.show()