2

我的问题是尝试使用每个元素都是 2 个值的元组的数组。

具体来说,问题是生成一个 200 步(但用于测试说 2)步的随机二维步行,最大距离,然后每个阶段 100 步(仅尝试 2 步),并从距前一阶段的起源。

我可以成功生成随机步骤数组并让它们返回最终位置 (x,y) 值,并计算它们与每次步行原点的距离:

这是在这些函数中定义的:

#............................................................getPositionInteger
def getPosInt(description) :
    """Asks the user to enter a positive integer"""
    askAgain = False
    while askAgain == False:
        try:
            posInt = eval(raw_input("\n %s :  " %description))
            assert 0 < posInt , "Not a positive integer"
            assert type(posInt) == int , "Not a positive integer"
            askAgain = True
        except :
            print "Input failed, not a positive integer"
    return posInt
#...............................................................initialPosition
def initialPosition() :
    """Initial position of walker at the start of a random walk"""
    return (0.0, 0.0)

#......................................................................distance
def distance(posA, posB) :
    """Distance between two positions"""
    xi = posA[0] ; yi = posA[1]
    xf = posB[0] ; yf = posB[1]
    return np.sqrt((xf-xi)**2+(yf-yi)**2)

    #..................................................................getPositions
def getPositions(start, nSteps, maxStep):
    xArray = maxStep * np.random.random(nSteps+1)* np.cos(2.0 * np.pi * random.random())
    yArray = maxStep * np.random.random(nSteps+1)* np.sin(2.0 * np.pi * random.random())
    xArray[0] = start[0]
    yArray[0] = start[-1]
    xArray = np.cumsum(xArray)
    yArray = np.cumsum(yArray)                                                    
    return (xArray[-1], yArray[-1])

但是我无法以每个阶段的 (x,y) 形式获得每个阶段每个步行的最终位置的数组

这是主要脚本以及我遇到问题的地方:

import numpy as np
import matplotlib.pylab as plt
import random
import time

MAX_STEP_SIZE = 0.90    # maximum size of a single step [m]
random.seed(12345)

#..........................................................................main
def main ():
    ''''''
    print "RANDOM WALK IN STAGES IN TWO DIMENSIONS"
    userdata = getDetails()
    print "\nPlease wait while random walks are generated and analyzed..."
    NUM_STAGES = userdata[0]
    NUM_WALKS = userdata[1]
    NUM_STEPS = userdata[2]
    stageStart = initialPosition()
    for stage in np.arange(NUM_STAGES):
        walks = np.zeros((NUM_WALKS, NUM_WALKS), dtype=np.ndarray)
        for walk in walks:
            walk = getPositions(stageStart, NUM_STEPS, MAX_STEP_SIZE)
            print walk
        print walks

您会看到我在制作 (x,y) 样式数组时遇到了麻烦,其中 [0 0] 应该是 [0.0 , 0.0] 并且打印了两次,此外,它不会更改为最终位置。

我非常感谢您提供的帮助、建议或参考资料。谢谢-席德

4

0 回答 0