Scipy DCT-II Pythona do 2D lub ND DCT

Question

Chciałbym użyć DCT-II scipy, ponieważ jest już zakodowany i szybki. Patrząc na doc, wydaje się, że jest to implementacja 1D. Czy można go używać w taki sposób, aby wykorzystać go jako implementację 3D? Nie jestem pewien co do matematyki. Czy implementacje 2D i 3D są równoważne z pomnożeniem 2 lub 3 razy 1D przy użyciu różnych wymiarów w obliczeniach?

Answer 1

Zasadniczo dodaje załatwia sprawę:

import numpy as np 
from scipy.fftpack import dct, idct 

# Lets create a 3D array and fill it with some values 
a = numpy.random.rand(3,3,3) 

f,x,y = a.shape 
b = np.zeros((f,x,y)) 

b = dct(dct(dct(a).transpose(0,2,1)).transpose(1,2,0)).transpose(1,2,0).transpose(0,2,1)