如何递归模拟随机游走?没有循环(Python)

时间:2014-09-28 17:01:46

标签: python recursion simulation random-walk

Python问题

我有一个随机步骤的功能:

def random_step(): 
    """ chooses a random step (-1 or 1) and returns it.
        inputs: none! However, make sure to use parens when calling it.
            For example: ramdom_step()
    """
    return random.choice([-1, 1])

我需要在我写的这个函数中调用它:

rw_outcome( start, numsteps ),需要两个输入:

  • start,一个表示梦游者起始位置的整数
  • numsteps,一个正整数,表示从起始位置开始的随机步骤数

它应模拟由numsteps随机步骤组成的随机游走,其大小是使用random_step()调用确定的,但我会继续返回相同的起始位置。

应该使用print返回的内容示例('start is',start):

>>> rw_outcome(40, 4)
start is 40
start is 41
start is 42
start is 41
start is 42
42

到目前为止我所拥有的:

def rw_outcome(start, numsteps):
    print('start is', start)
    if start + (numsteps*random_step()) == 0:
        return 0
    else:
        return rw_outcome(start,numsteps+1)

是否可以使用递归编写?

2 个答案:

答案 0 :(得分:5)

您的代码中存在一些错误。试试这个:

def rw_outcome(start, numsteps):
print('start is', start)
if numsteps == 0:
    return 0
else:
    return rw_outcome(start+random_step(),numsteps-1)

它应该有用。

答案 1 :(得分:0)

这样的东西?

def random_step(nsteps=1):
    if nsteps>0:
        return random_step(nsteps-1)+random.choice([-1, 1])
    else:
        return 0

def rw_outcome(start, numsteps):
    return start+random_step(numsteps)