0

我正在尝试编写一个函数，该函数采用网格中的行数和列数，模拟从网格中心开始的随机游走，并计算随机游走访问每个交叉点的次数。然后在随机游走移出网格后逐行打印表格到目前为止，我有这个，但我无法让它正常工作。

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]
4

2 回答 2

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和v表示交集，并且您已经初始化res为所有交集的二维列表，因此不需要循环。res我还将它移到了循环的顶部，否则您可能会在越过边界后尝试修改。

于 2013-06-04T17:30:46.527 回答
1

这是一个不那么冗长的版本，它使用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 + 麻木

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)
于 2013-06-04T19:27:54.170 回答