使用Cython时的lambda函数错误

6 投票
2 回答
2760 浏览
提问于 2025-04-16 07:42

我正在尝试使用Cython来加速一段代码。Cython给我报了一个错误,提示“期望一个标识符或字面量”,当我使用lambda函数的时候。根据我的了解,lambda函数在Cython 0.13版本中应该是被支持的。我是不是理解错了?如果它们确实被支持,我需要做些什么不同的事情来实现它们吗?

def f(e_1, e_2, rho):
    """Bivariate Normal pdf with mean zero, unit variances, and correlation coefficient rho."""
    return (1.0 / (2.0 * pi * sqrt(1 - rho**2))) * exp(-(1.0 / (2*(1 - rho**2))) * (e_1**2 + e_2**2 - 2*rho*e_1*e_2))

def P_zero(b_10, b_11, b_20, b_21, rho, gamma, x):
    """Returns the probability of observing zero entrants in a market by numerically
    integrating out the unobserved firm-specific profit shocks."""
    h_z = lambda e_1: -inf
    g_z = lambda e_1: -b_10 - b_11*x[0] - gamma*x[1]
    I   = lambda e_1, e_2: f(e_1, e_2, rho)
    return dblquad(I, -inf, (-b_20 - b_21*x[0] - gamma*x[2]), h_z, g_z)[0]

2 个回答

0

我可以在 OS X 10.6.6 上使用 Cython 0.14.1 编译并顺利运行下面这个简化的 Cython 代码。我不太清楚为什么在 0.13 版本上不行。最简单的解决办法就是升级 Cython,如果可以的话。

def f(e_1, e_2, rho):
    return e_1 + e_2 + rho

def dummy(a, b, c, d, e):
    return [a(1,2) + b + c + d(1) + e(3)]

def P_zero(b_10, b_11, b_20, b_21, rho, gamma, x):
    h_z = lambda e_1: -1000
    g_z = lambda e_1: -b_10 - b_11 * x[0] - gamma * x[1]
    I   = lambda e_1, e_2: f(e_1, e_2, rho)
    return dummy(I, -1000, (-b_20 - b_21 * x[0] - gamma * x[2]), h_z, g_z)[0]

print P_zero(1, 2, 3, 4, 5, 6, [6, 7, 8]) 
# outputs "-2122"
2

我觉得你应该把 h_z = lambda e_1: -inf 改成 h_z = lambda e_1: -float('inf'),除非你在别的地方定义过 inf

撰写回答