如何在Pythran中调用FFT？遇numpy.fft属性不支持问题求助

我之前在使用Pythran加速NumPy代码时也碰到过一模一样的问题——Pythran的静态分析确实不支持numpy.fft模块的相关函数，不过有个靠谱的解决方案，就是用Pythran官方推荐的FFTW库绑定来替代：

步骤1：安装FFTW3依赖 #

首先得在你的系统上安装FFTW3库，不同系统的安装命令如下：

• Ubuntu/Debian：sudo apt-get install libfftw3-dev
• macOS（Homebrew）：brew install fftw
• Windows：可以从FFTW官网下载预编译库，或者用conda安装conda install fftw

步骤2：修改代码使用Pythran的FFTW封装 #

把原来依赖numpy.fft.rfft的代码改成调用Pythran内置的pythran.fft模块，同时要在代码里添加编译指令告诉Pythran链接FFTW库：

import numpy as np
# pythran export testFunc(float[])
# pythran: include <fftw3.h>
# pythran: link fftw3
def testFunc(signal):
    from pythran.fft import rfft
    # 调用rfft，参数和numpy.fft.rfft一致
    X = rfft(signal, n=1024)
    return X

步骤3：带FFTW参数编译代码 #

编译的时候要加上-DUSE_FFTW参数，让Pythran启用FFTW支持：

pythran Foo.py -DUSE_FFTW

这样编译出来的代码不仅能实现和numpy.fft.rfft一样的功能，运行效率还会比原生NumPy的FFT更高，因为FFTW本身就是优化过的高性能FFT库。如果需要逆变换irfft，同样可以用pythran.fft.irfft，用法和NumPy版本完全一致。

内容的提问来源于stack exchange，提问作者jeanl