基于 Excel VBA 实现 DFT

最新推荐文章于 2022-04-07 19:22:14 发布

Patrick_PS

最新推荐文章于 2022-04-07 19:22:14 发布

阅读量1.4k

点赞数 2

文章标签： excel vba

本文链接：https://blog.csdn.net/weixin_45205765/article/details/110914722

版权

离散傅里叶变换（DFT）的计算公式如下：

DFT 将时域内的周期离散信号变换为频域内的离散周期信号。

1 实现的功能

利用 Excel VBA 编写了一个自定义的 DFT(Xn, Index=True) 函数，该函数返回数组。第一个输入参数是时域信号，要求在 Excel 表中以列的形式存在；第二个布尔型参数默认为 True，返回变换后的实部数组；取 False 时，返回虚部数组。

因为直接采用公式计算离散傅里叶变换，所以当数据量增多时，计算量较大。因此，处理很多数据时，建议利用其它工具。测试时，输入数据个数为720时，计算大概需要 3 秒（电脑型号较老，CPU为 Intel i5-3337U）。

2 使用方法

没有为函数增加许多判断，以应对不同的使用场景。使用具体步骤如下：

1) 输入数据为列的形式，如下图所示，选中 Data 列任意一个数据后，利用 Ctrl + ↓ 定位到最后一个数据，然后平移至 DFT VBA Real 列，随便输入一个数定位用。

2）按 Ctrl + ↑，回到 DFT VBA Real 列的第一个单元格（表头）。选中下一个单元格（B2），按 Ctrl + Shift + ↓，选中多个单元格（步骤1) 的作用在此体现），输入公式 =DFT(A2:A721)，其中 A2:A721 同样可以按 Ctrl + Shift + ↓ 选中。

3）按 Ctrl + Shift + Enter 进行数组运算（不能输花括号）。

3 具体代码

因为计算结果是对称的，所以只计算了前 N/2+1 个频率点，剩余的点在 Excel 表格中最后以 0 的形式出现（也可以增加循环，由前 N/2+1 的点的结果直接得到）。

Function DFT(Xn As Range, Optional Index As Boolean = True)
' 用公式直接计算 DFT
' 要求数据都是按列填入表格

PI = 3.141592653
Dim TempArray()

N = Xn.Count '统计输入的数据个数
ReDim TempArray(0 To N - 1) '动态改变数组长度

If Not Index Then '第二个参数输入 0 计算虚部，否则均计算实部

    For i = 0 To N / 2 '计算一半，另一半对称，取0处理
        temp = 0
        For j = 1 To N
            Angle = 2 * PI / N * (j - 1) * i
            temp = temp - Xn.Cells(j, 1).Value * Sin(Angle)
            'Xn对应单元格范围（一列数据），Cells()方法获得具体某个单元格，Value获得数值
        Next j
        TempArray(i) = temp / N '这里可以乘以2，得到单边谱
    Next i
    
    
Else '计算实部

    For i = 0 To N / 2
        temp = 0
        For j = 1 To N
            Angle = 2 * PI / N * (j - 1) * i
            temp = temp + Xn.Cells(j, 1).Value * Cos(Angle)
        Next j
        TempArray(i) = temp / N
    Next i
    
End If

DFT = Application.WorksheetFunction.Transpose(TempArray) '转置，变成列向量

End Function