如何在Julia中随机生成n阶离散傅里叶变换矩阵？

方法一：手动实现DFT矩阵 #

离散傅里叶变换（DFT）矩阵的核心定义为：n×n矩阵F的元素满足
$F[k,j] = \frac{1}{\sqrt{n}} e^{-2\pi i (k-1)(j-1)/n}$
其中$i$是虚数单位，k、j从1到n（适配Julia的1索引特性）。

直接用Julia实现该逻辑：

function dft_matrix(n::Int)
    ω = exp(-2πim / n)  # 计算n次单位根
    # 按公式生成矩阵元素
    F = [ω^((k-1)*(j-1)) / sqrt(n) for k in 1:n, j in 1:n]
    return F
end

# 示例：生成3×3的DFT矩阵
F3 = dft_matrix(3)

方法二：借助FFTW包生成 #

Julia的FFTW是高效傅里叶变换的标准库，我们可以通过对单位矩阵的每一列执行FFT来得到DFT矩阵（DFT矩阵作用于向量的结果等价于对向量做FFT）：

using FFTW

function dft_matrix_fftw(n::Int)
    # 创建n×n复数单位矩阵
    identity_mat = Matrix{ComplexF64}(I, n, n)
    # 对每一列做FFT后归一化
    F = fft(identity_mat, 1) / sqrt(n)
    return F
end

# 示例：生成4×4的DFT矩阵
F4 = dft_matrix_fftw(4)

补充说明 #

• 两种方法生成的矩阵完全等价，FFTW方法在n较大时性能更优；
• 若不需要正交归一化（即使用非正交DFT矩阵），只需移除代码中除以sqrt(n)的部分；
• 生成的矩阵元素均为复数类型，完全符合DFT的数学定义。

内容的提问来源于stack exchange，提问作者James Rider