使用 Cython 进行复值计算

Question

我正在尝试使用 cython 进行复数计算。在示例代码中，我想计算复数的复指数函数。问题是我不知道如何将我的整数乘以虚数单位。乘以 python 的虚数单位 1.0j 会在执行 cython 时引发错误。

这是我的代码：

cdef extern from "math.h":
    double complex cexp(double complex)

def testfunction():
    cdef double n
    n=3
    cdef double complex res
    res=cexp(n*1.0j)
    return res

这是错误消息：

complex.c:678:3: note: expected ‘complex double’ but argument is of type ‘__pyx_t_double_complex’

Answer 1

首先，尝试报告编译的完整堆栈跟踪。在我的机器上，我得到了一个非常有用的信息：

implicit declaration of function ‘cexp’ [-Wimplicit-function-declaration]

原来你输入的是错误的header。 cexp 在 <complex.h> 中声明，参见 docs.

只需将代码更改为

cdef extern from "complex.h":
    double complex cexp(double complex)

你会没事的。

使用 Cython 进行复值计算

Complex valued calculations using Cython

cython