python - 我可以安全地使用 np.random.binomial 和概率列表

Question

numpy.random.binomial(1, [.1, .5, .9])

似乎有效。

但是文档说第二个参数 p 是 float 而不是列表。

使用这个安全吗？

Answer 1

你说得对np.random.binomial's docstring说 p 预计是一个 float 。然而，该函数是defined here 。请注意，代码首先测试 p 是否为 float ，如果不是，则尝试将 p 转换为 NumPy 数组:

fp = PyFloat_AsDouble(p)
if not PyErr_Occurred():
    ...
    return discnp_array_sc(self.internal_state, rk_binomial, size, ln,
                           fp, self.lock)

PyErr_Clear()
...

op = <ndarray>PyArray_FROM_OTF(p, NPY_DOUBLE, NPY_ARRAY_ALIGNED)

因此，代码显然允许 p 可能是类似数组的可能性。