DCT 是一种线性变换,因此获取变换矩阵的一种方法是将其应用于单位矩阵。这是一个示例,我找到长度为 8 的序列的矩阵(将 8 更改为N
对于一般情况):
In [124]: import numpy as np
In [125]: from scipy.fft import dct
In [126]: D = dct(np.eye(8), axis=0)
D
是矩阵:
In [127]: D
Out[127]:
array([[ 2. , 2. , 2. , 2. , 2. , 2. , 2. , 2. ],
[ 1.96157056, 1.66293922, 1.11114047, 0.39018064, -0.39018064, -1.11114047, -1.66293922, -1.96157056],
[ 1.84775907, 0.76536686, -0.76536686, -1.84775907, -1.84775907, -0.76536686, 0.76536686, 1.84775907],
[ 1.66293922, -0.39018064, -1.96157056, -1.11114047, 1.11114047, 1.96157056, 0.39018064, -1.66293922],
[ 1.41421356, -1.41421356, -1.41421356, 1.41421356, 1.41421356, -1.41421356, -1.41421356, 1.41421356],
[ 1.11114047, -1.96157056, 0.39018064, 1.66293922, -1.66293922, -0.39018064, 1.96157056, -1.11114047],
[ 0.76536686, -1.84775907, 1.84775907, -0.76536686, -0.76536686, 1.84775907, -1.84775907, 0.76536686],
[ 0.39018064, -1.11114047, 1.66293922, -1.96157056, 1.96157056, -1.66293922, 1.11114047, -0.39018064]])
验证一下D @ x
相当于dct(x)
:
In [128]: x = np.array([1, 2, 0, -1, 3, 0, 1, -1])
In [129]: dct(x)
Out[129]: array([10. , 4.02535777, -1.39941754, 7.38025967, -1.41421356, -6.39104653, -7.07401092, 7.51550307])
In [130]: D @ x
Out[130]: array([10. , 4.02535777, -1.39941754, 7.38025967, -1.41421356, -6.39104653, -7.07401092, 7.51550307])
注意D @ x
通常会比dct(x)
.
为了与Matlab的完全一致dctmtx https://www.mathworks.com/help/images/ref/dctmtx.html,添加参数norm='ortho'
。例如,这里是dctmtx
在 Octave 中(返回与 Matlab 中相同的数组):
octave:1> pkg load image
octave:2> dctmtx(4)
ans =
0.50000 0.50000 0.50000 0.50000
0.65328 0.27060 -0.27060 -0.65328
0.50000 -0.50000 -0.50000 0.50000
0.27060 -0.65328 0.65328 -0.27060
这是使用的计算scipy.fft.dct
:
In [56]: from scipy.fft import dct
In [57]: dct(np.eye(4), axis=0, norm='ortho')
Out[57]:
array([[ 0.5 , 0.5 , 0.5 , 0.5 ],
[ 0.65328148, 0.27059805, -0.27059805, -0.65328148],
[ 0.5 , -0.5 , -0.5 , 0.5 ],
[ 0.27059805, -0.65328148, 0.65328148, -0.27059805]])