Python函数

Question

背景

在Julia编程中，我们有一个Rand函数，它是：

兰德(rng=GLOBAL_RNG，S，dims.)

它从S指定的一组值中选择一个随机元素或随机元素数组，并给出第三arg的维数数组。

问题

我想用同样的逻辑来模拟一个保证政策问题：

m = 60000
n = len(policy.AGE)
k=1/10
@time begin
    Death = zeros(Bool, m, n)
    Accident = rand(Bernoulli(k), m, n)

这是朱莉娅代码，但是当把它带到python时，我找不到rand(Bernoulli(k)，m，n)的正确形式。

我知道模拟伯努利只是bernoulli.rvs(k，n)，但是集合的值由S指定，并给出了第三arg.的维数数组。

Answer 1

您可能希望使用size关键字参数的rvs

>>> from scipy.stats import bernoulli
>>> bernoulli.rvs(0.25, size=10)
array([0, 0, 1, 0, 0, 0, 0, 0, 0, 0])