【Python学习笔记】21：numpy傅里叶变换

最新推荐文章于 2024-08-13 14:14:41 发布

大桔骑士v

最新推荐文章于 2024-08-13 14:14:41 发布

阅读量3.5w

点赞数 9

本文链接：https://blog.csdn.net/shu15121856/article/details/76154485

版权

Python 专栏收录该内容

48 篇文章 11 订阅

订阅专栏

一个简单的傅里叶变换

>>> import numpy as np
>>> import matplotlib.pyplot as plt
>>> x=np.linspace(0,2*np.pi,50)
>>> x
array([ 0.        ,  0.12822827,  0.25645654,  0.38468481,  0.51291309,
        0.64114136,  0.76936963,  0.8975979 ,  1.02582617,  1.15405444,
        1.28228272,  1.41051099,  1.53873926,  1.66696753,  1.7951958 ,
        1.92342407,  2.05165235,  2.17988062,  2.30810889,  2.43633716,
        2.56456543,  2.6927937 ,  2.82102197,  2.94925025,  3.07747852,
        3.20570679,  3.33393506,  3.46216333,  3.5903916 ,  3.71861988,
        3.84684815,  3.97507642,  4.10330469,  4.23153296,  4.35976123,
        4.48798951,  4.61621778,  4.74444605,  4.87267432,  5.00090259,
        5.12913086,  5.25735913,  5.38558741,  5.51381568,  5.64204395,
        5.77027222,  5.89850049,  6.02672876,  6.15495704,  6.28318531])
>>> wave=np.cos(x)
>>> transformed=np.fft.fft(wave) #傅里叶变换
>>> plt.plot(transformed) #绘制变换后的信号
Warning (from warnings module):
  File "D:\Program Files\Python\lib\site-packages\numpy\core\numeric.py", line 531
    return array(a, dtype, copy=False, order=order)
ComplexWarning: Casting complex values to real discards the imaginary part
[<matplotlib.lines.Line2D object at 0x000000000622FA20>]
>>> plt.show()

傅里叶变换使用np.fft中的fft函数，可以将余弦值作傅里叶变换。
不知道为什么按董老师的代码，绘制变换后的信号时会有这样的报错，不过最后的图像可以正常显示。
这里写图片描述

傅里叶反变换

>>> plt.plot(np.fft.ifft(transformed)) #反变换
[<matplotlib.lines.Line2D object at 0x00000000074F0630>]
>>> plt.show()

变换后的点用np.fft中的ifft就可以反变换回去，得到它的余弦的图像。
这里写图片描述

移频(针对作好傅里叶变换的数据)

>>> shifted=np.fft.fftshift(transformed) #移频
>>> plt.plot(shifted)
[<matplotlib.lines.Line2D object at 0x00000000078AF7B8>]
>>> plt.show()

使用np.fft中的ffshift可以对信号进行移频操作，还有iffshift可以将移频后的信号还原成之前的。
这里写图片描述

对移频后的信号进行傅里叶反变换

>>> plt.plot(np.fft.ifft(shifted))
[<matplotlib.lines.Line2D object at 0x0000000007918DD8>]
>>> plt.show()

这里写图片描述

二维傅里叶变换

>>> import numpy as np
>>> import matplotlib.pyplot as plt
>>> x=np.random.rand(10,10) #二维随机信号
>>> wave=np.cos(x)
>>> plt.plot(wave)
>>> plt.show()

这里写图片描述

>>> transformed=np.fft.fft2(wave)
>>> plt.plot(transformed)
>>> plt.show()

二维傅里叶变换使用的函数是np.fft里的fft2()函数。
这里写图片描述

>>> plt.plot(np.fft.ifft2(transformed)) #二维反变换
>>> plt.show()

同样地二维反变换也使用ifft2()函数。
这里写图片描述
得到的图像和最开始变换前的图像理论上是一样的。

滤出图像中的低频信号

首先先安装python3下的PIL包，用pip安装pillow模块就可以了。但在这之前可能需要对pip更新。
这里写图片描述
等待安装完成(cmd会回到之前的文件夹)，就可以使用PIL包了。pillow模块用于图像处理非常强大，其中的Image类是比较常用的一个类。
选择一个图像作为测试，如下图。

import numpy as np
from PIL import Image
from numpy.fft import fft,ifft
def filterImage(srcImage):
    #打开图像文件并获取数据
    srcIm=Image.open(srcImage)
    srcArray=np.fromstring(srcIm.tobytes(),dtype=np.int8)
    #傅里叶变换并滤除低频信号
    result=fft(srcArray)
    result=np.where(np.absolute(result)<9e4,0,result)
    #傅里叶反变换,保留实部
    result=ifft(result)
    result=np.int8(np.real(result))
    #转换为图像
    im=Image.frombytes(srcIm.mode,srcIm.size,result)
    im.show()
    
filterImage('test.jpg')