9阅网

您现在的位置是:首页 > 知识 > 正文

知识

python - Python多线程随机生成

admin2022-10-31知识20

我试图在我的模拟中实现这段代码。

https:/numpy.orgdocstablereferencerandommultithreading.html。

但我不能解决这个问题。

如果我按照链接中的例子,我得到的是

mrng = MultithreadedRNG(10000000, seed=0)
mrng.fill()
print(mrng.values[-1])
> 0.0

而其他所有的值也都是0。

如果我给一个较小的输入数字,如40,我得到的是

mrng = MultithreadedRNG(40)
mrng.fill()
print(mrng.values[-1])
> array([1.08305179e-311, 1.08304781e-311, 1.36362118e-321,             nan,
       6.95195359e-310, ...., 7.27916164e-095, 3.81693953e+180])

我做错了什么?我只是想把这个多处理代码实现到一个随机比特(0 1)发生器上。



【回答】:

我想,这个例子中存在一个错误。你必须把 PCG64 包入 Generator 界面。

试试下面的代码

class MultithreadedRNG(object):
    def __init__(self, n, seed=None, threads=None):
        rg = PCG64(seed)
        if threads is None:
            threads = multiprocessing.cpu_count()
        self.threads = threads

        self._random_generators = [Generator(rg)]
        last_rg = rg
        for _ in range(0, threads-1):
            new_rg = last_rg.jumped()
            self._random_generators.append(Generator(new_rg))
            last_rg = new_rg

        self.n = n
        self.executor = concurrent.futures.ThreadPoolExecutor(threads)
        self.values = np.empty(n)
        self.step = np.ceil(n / threads).astype(np.int_)

    def fill(self):
        def _fill(gen, out, first, last):
            gen.standard_normal(out=out[first:last])

        futures = {}
        for i in range(self.threads):
            args = (_fill,
                    self._random_generators[i],
                    self.values,
                    i * self.step,
                    (i + 1) * self.step)
            futures[self.executor.submit(*args)] = i
        concurrent.futures.wait(futures)

    def __del__(self):
        self.executor.shutdown(False)

没怎么测试过,但数值看起来还可以。