一维阵列的快速傅里叶变换

Question

我有一个简单的一维数组，如[0,0,0,0,0,1,1,1,1,1,0,0,0,0,0]，它描述了一个平方脉冲。我想把这个脉冲变换到频域，并通过使用下面的代码绘制它的幅度谱(我从OpenCV Python教程中得到的):

squareimpulse = np.array([0,0,0,0,0,1,1,1,1,1,0,0,0,0,0])
img = (squareimpulse)
f = np.fft.fft(img)
fshift = np.fft.fftshift(f)
magnitude_spectrum = 20*np.log(np.abs(fshift))
plt.subplot(121),plt.imshow(img, cmap = 'gray')
plt.title('Input Image'), plt.xticks([]), plt.yticks([])
plt.subplot(122),plt.imshow(magnitude_spectrum, cmap = 'gray')
plt.title('Magnitude Spectrum'), plt.xticks([]), plt.yticks([])
plt.show()

脚本工作良好的二维数组与f = np.fft.fft2(img)，但不适合我的情况下，我只有一个维度。

Answer 1

plt.magnitude_spectrum(img)
plt.show()

好吧，说真的，imshow函数不接受一个值列表。看到http://matplotlib.org/api/pyplot_api.html matplotlib.pyplot.imshow

Answer 2

我将plt.imshow(.....)更改为plt.plot(.....)，脚本现在正在工作!

import cv2
import numpy as np
from matplotlib import pyplot as plt
    squareimpulse = np.array([0,0,0,0,0,1,1,1,1,1,0,0,0,0,0])
    img = (squareimpulse)
    f = np.fft.fft(img)
    fshift = np.fft.fftshift(f)
    magnitude_spectrum = (np.abs(fshift))
    plt.subplot(121)
    plt.plot(img)
    plt.title('Input Image')
    plt.xticks([]), plt.yticks([])
    plt.subplot(122)
    plt.plot(magnitude_spectrum)
    plt.title('Magnitude Spectrum')
    plt.xticks([]), plt.yticks([])
    plt.show()