fourier-numpy.py

# Laskee input-matriisin Fast Fourier muunnoksen FFT
# ja tuloksesta käänteismuunnoksen IFFT,
# jolloin pitäisi tulla alkuperäinen input-matriisi
# Juhani Kaukoranta 12.4.2019
import numpy as np
import time
n = int(input("Anna n*n matriisin dimensio n "))
A = np.random.rand(n,n)
time0 = time.perf_counter()
X = np.fft.fftn(A)
Y = np.fft.ifftn(X)
time1 = time.perf_counter()
print("\n Aikaa FFT-muunnokseen ja käänteismuunnokseen kului ",time1-time0," sekuntia")
#print("input-matriisi A: \n",A)
#print("A:n FFT-muunnos \n",X)
#print("käänteismuunnos IFFT, pitäisi olla alkuperäinen matriisi A \n",Y)