Python,SimPy:在函数内部使用yield

时间:2011-11-28 13:59:44

标签: python simulation simpy

Helo,我正在SimPy中构建一个相对复杂的离散事件仿真模型。

当我尝试将yield语句放在函数中时,我的程序似乎不起作用。下面是一个例子。

import SimPy.SimulationTrace as Sim
import random

## Model components ##
class Customer(Sim.Process):
    def visit(self):
        yield Sim.hold, self, 2.0
        if random.random()<0.5:
            self.holdLong()
        else:
            self.holdShort()

    def holdLong(self):
        yield Sim.hold, self, 1.0
        # more yeild statements to follow

    def holdShort(self):
        yield Sim.hold, self, 0.5
        # more yeild statements to follow

## Experiment data ##
maxTime = 10.0 #minutes

## Model/Experiment ##
#random.seed(12345)
Sim.initialize()
c = Customer(name = "Klaus") #customer object
Sim.activate(c, c.visit(), at = 1.0)
Sim.simulate(until=maxTime)

我从运行中获得的输出是:

0 activate <Klaus > at time: 1.0 prior: False
1.0 hold  < Klaus >  delay: 2.0
3.0 <Klaus > terminated

holdLong()和holdShort方法似乎根本不起作用。我怎样才能解决这个问题?提前谢谢。

2 个答案:

答案 0 :(得分:6)

调用生成器函数返回可以迭代的生成器对象。你只是忽略了这个返回值,所以没有任何反应。相反,您应该迭代生成器并重新生成所有值:

class Customer(Sim.Process):
    def visit(self):
        yield Sim.hold, self, 2.0
        if random.random()<0.5:
            for x in self.holdLong():
                yield x
        else:
            for x in self.holdShort():
                yield x

答案 1 :(得分:1)

在Python中,yield不能通过函数调用向上传播。将visit更改为以下内容:

def visit(self):
    yield Sim.hold, self, 2.0
    if random.random()<0.5:
        for x in self.holdLong():
            yield x
    else:
        for x in self.holdShort():
            yield x