加载其他参数到scipy的rv_continuous

问题描述

我打算对连续的,非分析性的随机变量进行一些基本的代数运算。我想将它们的概率密度函数定义为数组x和f(x)。

但是,我很惊讶地发现似乎没有任何软件包可以执行基本操作,例如计算两个pdf的总和或乘积分布(如果我错了,请纠正我)。为了由我自己执行这些操作,然后我计划在this线程之后创建scipy.stats rv_continuous的子类:

import scipy as sp
import numpy as np

class my_pdf(sp.stats.rv_continuous):
    def __init__(self,x,p):
        self.x = x
        self.p = p

    def _pdf(self,x):
        return sp.interpolate.interp1d(self.x,self.p)(x)

x = np.linspace(0,1,101)
f = 3*x**2
my_cv = my_pdf(x,f)
my_cv.pdf(0)

最后一行引发错误,因为覆盖init方法可能不是解决方法。除了从头开始构建所有内容之外,是否有办法将其他参数传递给rv_continuous或解决该问题的另一种方式?

解决方法

这似乎对我有用:

import scipy as sp
import scipy.stats
import numpy as np

class my_pdf(sp.stats.rv_continuous):
    def __init__(self,x,p):
        super().__init__(a=x.min(),b=x.max())
        self.x = x
        self.p = p
    
    def _pdf(self,x):
        return sp.interpolate.interp1d(self.x,self.p)(x)

x = np.linspace(0,1,101)
f = 3*x**2
my_cv = my_pdf(x,f)
my_cv.pdf(0)
my_cv.cdf(0.5)