有没有一种更有效的方法来枚举python或R中离散随机变量的每个可能结果的概率？

作者: 不见你
发布时间: 2025-01-04 12:14:41 (28天前)
转自：

3 条回复

0#
回复此人
哇塞 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 这是一个 <code> fft </code> 基于 <code> numpy </code> 解： </p> <pre> <code> import numpy as np from scipy import fftpack def toss(n=10, p=0.5): t1 = np.zeros(fftpack.next_fast_len(n+1)) t1[:2] = 1-p, p f1 = fftpack.rfft(t1) c1 = f1[1:(len(t1) - 1) // 2 * 2 + 1].view(f'c{2*t1.itemsize}') c1 **= n f1[::(len(t1) + 1) // 2 * 2 - 1] **= n return fftpack.irfft(f1)[:n+1] </code> </pre> <P> 例如： </p> <pre> <code> >>> toss(3) array([0.125, 0.375, 0.375, 0.125]) >>> toss(10) array([0.00097656, 0.00976562, 0.04394531, 0.1171875 , 0.20507813, 0.24609375, 0.20507813, 0.1171875 , 0.04394531, 0.00976562, 0.00097656]) </code> </pre> </DIV>

编辑
1#
回复此人
庸人自扰1 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 使用Python标准库，您可以将概率作为有理数（这是精确解），例如 </p> <pre> <code> from fractions import Fraction from math import factorial n=30 [Fraction(factorial(n), factorial(n - j)) * Fraction(1, factorial(j) * 2 ** n) for j in range(0, n + 1)] </code> </pre> <P> 这可以很容易地转换成浮点数，例如 </p> <pre> <code> list(map(float, [Fraction(factorial(n), factorial(n - j)) * Fraction(1, factorial(j) * 2 ** n) for j in range(0, n + 1)])) </code> </pre> </DIV>

编辑

登录后才能参与评论