Question

我正在尝试使用C numpy.fft.fft库重现numpy.fft.fft2和FFTW的输出。

>>> b
array([1, 2, 3, 4, 5, 6])
>>> type(b)
<class 'numpy.ndarray'>
>>> b.shape
(6,)
>>> np.fft.fft(b)
array([21.+0.j        , -3.+5.19615242j, -3.+1.73205081j, -3.+0.j        ,
       -3.-1.73205081j, -3.-5.19615242j])

此输出可以通过以下方式获得：

int N = 10;
double in[] = {1,2,3,4,5,6,0,0,0,0};
fftw_complex *out;
fftw_plan p;
out = (fftw_complex*)fftw_malloc(sizeof(fftw_complex) * (N/2 +1));
p = fftw_plan_dft_r2c_1d(6, in, out, FFTW_ESTIMATE);
fftw_execute(p);
fftw_destroy_plan(p);
fftw_free(out);

类似地，可以复制传递到numpy.fft.fft2的2d数组的输出：

>>> a
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> a.shape
(3, 2)
>>> np.fft.fft2(a)
array([[21.+0.j        , -3.+0.j        ],
       [-6.+3.46410162j,  0.+0.j        ],
       [-6.-3.46410162j,  0.+0.j        ]])

，相应的C ++代码为（仅一行更改）

p = fftw_plan_dft_r2c_2d(3, 2, in, out, FFTW_ESTIMATE);

我遇到了将2d array传递到numpy.fft.fft的Python代码

>>> a
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> a.shape
(3, 2)
>>> np.fft.fft(a)
array([[ 3.+0.j, -1.+0.j],
       [ 7.+0.j, -1.+0.j],
       [11.+0.j, -1.+0.j]])

我正在尝试找出如何使用FFTW APIs实现这一目标。关于如何重现这一点的任何线索？还是为什么numpy允许矩阵2d数组的1D Fourier transformation？

Answer 1

为什么NumPy允许将二维数组传递给一维FFT？目的是能够同时计算多个单个一维信号的FFT。

如果

>>> a = np.array([[1, 2], [3, 4], [5, 6]])
>>> A = np.fft.fft(a)

然后，A的第一行将是a的第一行的一维FFT。 A的第二行是a的第二行的一维FFT，依此类推。

可以通过

进行验证

>>> np.fft.fft(a[0, :])
array([ 3.+0.j, -1.+0.j])

>>> A[0, :]
array([ 3.+0.j, -1.+0.j])

或

>>> np.fft.fft(a[1, :])
array([ 7.+0.j, -1.+0.j])

>>> A[1, :]
array([ 7.+0.j, -1.+0.j])

要在FFTW中执行相同的操作，可以对不同的行多次执行计划，也可以使用fftw_plan fftw_plan_many_dft。

二维数组的numpy.fft numpy.fft2和FFTW

1 个答案: