纯Python

Question

我正在尝试编写一个函数，它接受网格中的数字或行和列，模拟从网格中心开始的随机游走，并计算随机游走访问每个交叉点的次数。然后，当随机游走移动到网格外时，逐行打印表格到目前为止，我有这个，但我不能让它正常工作。

def manhattan(x,y):
    'int,int==>nonetype'
    import random
    res=[]
    for i in range(x):
        res.append([])
    for i in res:
        for j in range(y):
            i.append(0)
    position=(x//2+1,y//2+1)
    z=position[0]
    v=position[1]

    while z!=-1 or z!=x or v!=-1 or v!=y:
        direction=random.randrange(1,5)
        if direction==1:
            v+=1
        elif direction==2:
            z+=1
        elif direction==3:
            v-=1
        else:
            z-=1
        for i in range(len(res)):
            if i ==z:
                res[i]+=1
            for j in range(i):
                if v==j:
                    i[j]+=1
    for i in res:
        print(i)

完成后应阅读：

manhattan(5,11)
[0,0,0,0,0,0,0,0,0,0,0]
[0,0,0,0,0,0,0,0,0,0,0]
[0,0,0,0,0,1,1,1,1,2,2]
[0,0,0,0,0,0,0,0,0,0,0]
[0,0,0,0,0,0,0,0,0,0,0]

Answer 1

你非常接近，请尝试以下方法：

def manhattan(x,y):
    'int,int==>nonetype'
    import random
    res=[]
    for i in range(x):
        res.append([])
    for i in res:
        for j in range(y):
            i.append(0)
    position=(x//2+1,y//2+1)
    z=position[0]
    v=position[1]

    while z!=-1 and z!=x and v!=-1 and v!=y:
        res[z][v] += 1
        direction=random.randrange(1,5)
        if direction==1:
            v+=1
        elif direction==2:
            z+=1
        elif direction==3:
            v-=1
        else:
            z-=1
    for i in res:
        print(i)

在while循环之前没有什么不同，并且只有一些变化。首先，您需要在循环条件检查中使用and而不是or，因为如果满足任何条件，您就要退出。

另一项更改是从for循环底部删除while循环，并将其替换为res[z][v] += 1，这可行，因为z和{{1} }表示交集，并且您已将v初始化为所有交叉点的二维列表，因此不需要循环。我也把它移到了循环的顶部，否则你可能会在移过边界后尝试修改res。

Answer 2

这是一个使用random.choice而不是链式elif语句的详细版本。我发现学习python以不同的方式看到同样的问题很有帮助，所以这里有一个纯python和一个numpy + python实现。

纯Python

import random

def manhattan(n,m):
    grid = [[0,]*m for _ in xrange(n)]
    directions = [[-1,0],[1,0],[0,-1],[0,1]]
    pt = [n//2, m//2]

    while pt[0]>=0 and pt[0]<n and pt[1]>=0 and pt[1]<m:
        grid[pt[0]][pt[1]] += 1
        d  = random.choice(directions)
        pt[0] += d[0]
        pt[1] += d[1]
    return grid

for row in manhattan(5,11):
    print row

例如，这提供了

[0, 0, 0, 1, 3, 3, 0, 0, 0, 0, 0]
[0, 0, 0, 0, 1, 2, 0, 0, 0, 0, 0]
[0, 0, 0, 0, 1, 3, 3, 2, 0, 0, 0]
[0, 0, 0, 0, 0, 1, 2, 2, 1, 0, 0]
[0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0]

Python + numpy

import numpy as np
import random

def manhattan(n,m):
    grid = np.zeros((n,m),dtype=int)
    directions = [[-1,0],[1,0],[0,-1],[0,1]]
    pt   = np.array([n//2, m//2])

    while (pt>=0).all() and (pt<grid.shape).all():
        grid[pt[0],pt[1]]  += 1      
        pt += random.choice(directions)
    return grid

print manhattan(5,11)

随机漫步的曼哈顿网格方向

2 个答案:

纯Python

Python + numpy