如何在kstest中使用可调用函数？

Question

def Ftrapz(x):
    global a2
    if x<a:
        a2=0
    elif a<=x<=b:
        a2=(0.5*k1*x**2+b1*x)-(0.5*k1*a**2+b1*a)
    else:
        a2=(0.5*k1*b**2+b1*b)-(0.5*k1*a**2+b1*a)
    return a2

trapz_test=stats.kstest(x0,Ftrapz)

我想使用stats.kstest测试我自己定义的数组x0和分布函数。 但是这个程序有一些错误，那么如何正确使用可调用函数呢？ 据说：

ValueError：具有多个元素的数组的真值不明确。 使用a.any（）或a.all（）

Answer 1

kstest方法将NumPy数组作为参数传递给Ftrapz。 因此，在您的if语句中， x<a是布尔数组，这会导致有关歧义性的错误。 解决方案不是使用all或any ，而是重写该函数以使其处理数组。 一种典型的实现方法是使用numpy.piecewise ，它获取一个范围列表和要在这些范围内执行的函数列表（它们可能是恒定的）。 您的函数可以表示为：

def Ftrapz(x):
    ranges = [x < a, np.logical_and(a <= x, x <= b), x > b]
    values = [0, lambda x : (0.5*k1*x**2+b1*x)-(0.5*k1*a**2+b1*a), (0.5*k1*b**2+b1*b)-(0.5*k1*a**2+b1*a)]
    return np.piecewise(x, ranges, values)

请注意，NumPy数组的比较不能像标量那样方便地嵌套： a <= x <= b会引发错误。 通过将相应的公式logical_and最后一个公式，可以避免混乱的logical_and ，并在其他条件失败时执行该公式：

    ranges = [x < a, x > b]
    values = [0, (0.5*k1*b**2+b1*b)-(0.5*k1*a**2+b1*a), lambda x : (0.5*k1*x**2+b1*x)-(0.5*k1*a**2+b1*a)]