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

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

用法:

fft.fftshift(x, axes=None)

将zero-frequency 分量移动到频谱的中心。

此函数为列出的所有轴交换half-spaces(默认为全部)。请注意，仅当 len(x) 为偶数时，y[0] 才是奈奎斯特分量。

参数：

x： array_like

输入数组。

axes： int 或形状元组，可选

要移动的轴。默认为无，这会移动所有轴。

返回：

y： ndarray

移位的数组。

例子：

>>> freqs = np.fft.fftfreq(10, 0.1)
>>> freqs
array([ 0.,  1.,  2., ..., -3., -2., -1.])
>>> np.fft.fftshift(freqs)
array([-5., -4., -3., -2., -1.,  0.,  1.,  2.,  3.,  4.])

仅沿第二个轴移动zero-frequency 组件：

>>> freqs = np.fft.fftfreq(9, d=1./9).reshape(3, 3)
>>> freqs
array([[ 0.,  1.,  2.],
       [ 3.,  4., -4.],
       [-3., -2., -1.]])
>>> np.fft.fftshift(freqs, axes=(1,))
array([[ 2.,  0.,  1.],
       [-4.,  3.,  4.],
       [-1., -3., -2.]])