Случайно генерируемые прогулки на Python дают тот же результат - PullRequest
Новая
       27

Случайно генерируемые прогулки на Python дают тот же результат

0 голосов
/ 02 мая 2018 
from pylab import *

no_steps = 10000

number = random()
position = zeros(no_steps)
position[0] = 0
time = zeros(no_steps)
time[0] = 0

for i in range(1, no_steps):
    time[i] = time[i-1] + 1
    if number >= 0.5:
        position[i] = position[i-1] + 1
        number = random()
    else:
        position[i] = position[i-1] - 1
        number = random()
plot(time, position)

number2 = random()
position2 = zeros(no_steps)
position2[0] = 0
time2 = zeros(no_steps)
time2[0] = 0
for t2 in range(1, no_steps):
    time2[t2] = time[t2-1] + 1
    if number2 >= 0.5:
        position2[t2] = position2[t2-1] + 1
        number2 = random()
    else:
        position2[t2] = position[t2-1] - 1
        number2 = random()
plot(time2,position2)

Это должно генерировать случайные прогулки, генерируя случайное число каждый раз и проверяя условия. Поэтому я предположил, что если это работает на одну прогулку, я могу просто добавить больше одинаковых и поместить их все в один график в конце. Тем не менее, очевидно, что это не так, и графики, которые в итоге построены, чрезвычайно похожи, с разницей в позициях по некоторым причинам, равной -2. Код, если я запускаю блоки отдельно от их собственной программы, сгенерирует два совершенно разных варианта, только когда я их соберу, он перестанет работать так, как задумано. Что именно мне не хватает?

1 Ответ

0 голосов
/ 02 мая 2018

Вы случайно повторно использовали переменные с первого графика: 

for t2 in range(1, no_steps):
    time2[t2] = time[t2-1] + 1
    ^^^^^       ^^^^
    if number2 >= 0.5:
        position2[t2] = position2[t2-1] + 1
        number2 = random()
    else:
        position2[t2] = position[t2-1] - 1
        ^^^^^^^^^       ^^^^^^^^
        number2 = random()
plot(time2,position2)

Я бы сгенерировал случайное блуждание с помощью функции, чтобы вам не приходилось беспокоиться о переименовании переменных следующим образом: 

import numpy
from pylab import *

no_steps = 10000

def random_walk(no_steps):
    # 2 * [0, 1] - 1  ->  [0, 2] - 1  ->  [-1, 1]
    directions = 2 * numpy.random.randint(0, 2, size=(1, no_steps)) - 1
    positions = numpy.cumsum(directions)
    positions -= positions[0]  # To make it start from zero

    return positions

time1 = numpy.arange(0, no_steps)
plot(time1, random_walk(no_steps))
savefig('1.png')

clf()

time2 = numpy.arange(0, no_steps)
plot(time2, random_walk(no_steps))
savefig('2.png')
...