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.]])