Python numpy fft.rfftn用法及代码示例

本文简要介绍 python 语言中 numpy.fft.rfftn 的用法。

用法:

fft.rfftn(a, s=None, axes=None, norm=None)

计算实际输入的 N 维离散傅里叶变换。

此函数通过快速傅里叶变换 (FFT) 在 M-dimensional 实数数组中的任意数量的轴上计算 N 维离散傅里叶变换。默认情况下，所有轴都被变换，实际变换在最后一个轴上执行，而其余变换是复杂的。

参数：

a： array_like

输入数组，被认为是真实的。

s： 整数序列，可选

要从输入中使用的形状(沿每个变换轴的长度)。 (s[0]指轴 0，s[1]到轴 1 等)。最终元素为s对应于n为了rfft(x, n)，而对于其余的轴，它对应于n为了fft(x, n).沿着任何轴，如果给定的形状小于输入的形状，则裁剪输入。如果它更大，则用零填充输入。如果s未给出，输入沿指定轴的形状轴被使用。

axes： 整数序列，可选

计算 FFT 的轴。如果没有给出，最后一个len(s)使用轴，或所有轴，如果s也没有指定。

norm： {“backward”, “ortho”, “forward”}，可选

标准化模式(参见 numpy.fft )。默认为“backward”。指示前向/后向变换对的哪个方向被缩放以及使用什么归一化因子。

返回：

out： 复杂的ndarray

截断或补零的输入，沿由轴，或通过组合s和a，如上面参数部分所述。最后变换的轴的长度将是s[-1]//2+1，而剩余的变换轴的长度将根据s，或与输入保持不变。

抛出：

ValueError

如果 s 和轴有不同的长度。

IndexError

如果轴的元素大于a的轴数。

注意：

实际输入的变换在最后一个变换轴上执行，如 rfft ，然后在其余轴上的变换如 fftn 。输出的顺序与最后一个变换轴的 rfft 相同，其余变换轴的输出顺序与 fftn 相同。

有关使用的详细信息、定义和约定，请参阅 fft 。

例子：

>>> a = np.ones((2, 2, 2))
>>> np.fft.rfftn(a)
array([[[8.+0.j,  0.+0.j], # may vary
        [0.+0.j,  0.+0.j]],
       [[0.+0.j,  0.+0.j],
        [0.+0.j,  0.+0.j]]])

>>> np.fft.rfftn(a, axes=(2, 0))
array([[[4.+0.j,  0.+0.j], # may vary
        [4.+0.j,  0.+0.j]],
       [[0.+0.j,  0.+0.j],
        [0.+0.j,  0.+0.j]]])