r - 新的概率分布几何分布，均值为 1/p

Question

我想声明一个新的概率分布，我需要一个几何分布，我知道在 R 中已经声明了这种类型的分布，但我需要 p(1-p)^(k-1) 形式的几何分布，所以平均值是 1/p 并且 k={1, 2, ...}。然后我想使用 rgeom 等类型的所有功能。

非常感谢您的帮助。

标签： r

Answer 1

R 中的几何分布被定义为p(1-p)^(x)不包括第一次成功的试验次数。

要获得包括第一次成功在内的所有试验的分布，您可以简单地相应地调整公式。

my_rgeom <- function(n, prob) rgeom(n, prob) + 1
my_dgeom <- function(x, prob, log = FALSE) {
  p <- dgeom(x, prob) / (1 - prob)
  if (isTRUE(log)) log(p) else p
}