python-3.x - Python中二项分布的概率质量函数

我正在使用Python3计算此维基百科示例的概率质量函数(PMF):



我试图遵循以下scipy文档:

https://docs.scipy.org/doc/scipy-0.19.0/reference/generated/scipy.stats.binom.html

该文档明确指出:

Notes

The probability mass function for binom is:

binom.pmf(k) = choose(n, k) * p**k * (1-p)**(n-k)

for k in {0, 1,..., n}.

binom takes n and p as shape parameters.

好吧，我在考虑维基百科示例的情况下尝试实现了这一点。这是我的代码:

from scipy.stats import binom

n = 6

p = 0.3

binom.pmf(k) = choose(n, k) * p**k * (1-p)**(n-k)

print  (binom.pmf(1))

但是，我收到此错误的消息:

  File "binomial-oab.py", line 7
    binom.pmf(k) = choose(n, k) * p**k * (1-p)**(n-k)
    ^
SyntaxError: can't assign to function call

我该如何解决？

最佳答案

只需调用binom.pmf(1, n, p)即可获得k=1的结果。文档中的表达式只是向您显示数学上如何定义PMF，而不是您希望执行的实际代码段。