Как мне инициализировать график моей функции, чтобы начать с 0? - PullRequest
Новая
       28

Как мне инициализировать график моей функции, чтобы начать с 0?

0 голосов
/ 23 ноября 2018

Это, наверное, действительно глупый вопрос, но я просто не могу понять, как это сделать по какой-то причине.Здесь я создал функцию для случайного блуждания, которая просто использует двоичную биномиальную функцию с одной пробой (т. Е. Если она меньше 0,5, то она равна -1, больше ее +1. Однако это, очевидно, делает первое значение функции либо -1, либо+1, тогда как я хочу, чтобы оно было равным нулю. Как я могу это изменить? 

import numpy as np
import matplotlib.pyplot as plt

def random_walk(N,d):
    walk = d*np.cumsum(2*np.random.binomial(1,.5,N-1)-1)
    return walk

plt.plot(np.arange(1,250),random_walk(250,1))
plt.show()

Опять же, это, вероятно, очень просто, и я действительно глуп, но я очень ценю помощь!

1 Ответ

0 голосов
/ 23 ноября 2018

Небольшая настройка вашего кода.Надеюсь, это то, что вы ищете. 

import numpy as np
import matplotlib.pyplot as plt

def random_walk(N,d):
    walk = np.concatenate(([0],np.cumsum(2*np.random.binomial(1,.5,N-1)-1)))
    return walk

plt.plot(np.arange(0,250),random_walk(250,1))
plt.show()
...