一维数组的Numpy FFT（快速傅立叶变换）

Question

我有一个简单的一维数组，如[0,0,0,0,0,1,1,1,1,1,0,0,0,0,0]，它描述了一个方形脉冲。我想这种冲动转化到频域，并通过使用下面的代码绘制其幅度谱（我是从OpenCV Python Tutorials）：

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

Answer 2

plt.magnitude_spectrum(img) 
plt.show() 

嗯，严重的是，imshow函数不接受只是一个值列表。请参阅http://matplotlib.org/api/pyplot_api.html#matplotlib.pyplot.imshow