Révision 300 ETSN/MyDFT_3.py
| MyDFT_3.py (revision 300) | ||
|---|---|---|
| 11 | 11 |
Y=np.zeros(size).astype(np.float32) |
| 12 | 12 |
for i in range(size): |
| 13 | 13 |
for j in range(size): |
| 14 |
X[i]=X[i]+x[j]*cos(2.*pi*i*j/size)-y[j]*sin(2.*pi*i*j/size)
|
|
| 15 |
Y[i]=Y[i]+x[j]*sin(2.*pi*i*j/size)+y[j]*cos(2.*pi*i*j/size)
|
|
| 14 |
X[i]=X[i]+x[j]*cos(2.*pi*i*j/size)+y[j]*sin(2.*pi*i*j/size)
|
|
| 15 |
Y[i]=Y[i]-x[j]*sin(2.*pi*i*j/size)+y[j]*cos(2.*pi*i*j/size)
|
|
| 16 | 16 |
return(X,Y) |
| 17 | 17 |
|
| 18 | 18 |
# Numpy Discrete Fourier Transform |
| ... | ... | |
| 22 | 22 |
Y=np.zeros(size).astype(np.float32) |
| 23 | 23 |
nj=np.multiply(2.0*np.pi/size,np.arange(size)).astype(np.float32) |
| 24 | 24 |
for i in range(size): |
| 25 |
X[i]=np.sum(np.subtract(np.multiply(np.cos(i*nj),x),np.multiply(np.sin(i*nj),y)))
|
|
| 26 |
Y[i]=np.sum(np.add(np.multiply(np.sin(i*nj),x),np.multiply(np.cos(i*nj),y)))
|
|
| 25 |
X[i]=np.sum(np.add(np.multiply(np.cos(i*nj),x),np.multiply(np.sin(i*nj),y)))
|
|
| 26 |
Y[i]=np.sum(np.subtract(np.multiply(np.cos(i*nj),y),np.multiply(np.sin(i*nj),x)))
|
|
| 27 | 27 |
return(X,Y) |
| 28 | 28 |
|
| 29 | 29 |
# Numba Discrete Fourier Transform |
| ... | ... | |
| 35 | 35 |
Y=np.zeros(size).astype(np.float32) |
| 36 | 36 |
nj=np.multiply(2.0*np.pi/size,np.arange(size)).astype(np.float32) |
| 37 | 37 |
for i in numba.prange(size): |
| 38 |
X[i]=np.sum(np.subtract(np.multiply(np.cos(i*nj),x),np.multiply(np.sin(i*nj),y)))
|
|
| 39 |
Y[i]=np.sum(np.add(np.multiply(np.sin(i*nj),x),np.multiply(np.cos(i*nj),y)))
|
|
| 38 |
X[i]=np.sum(np.add(np.multiply(np.cos(i*nj),x),np.multiply(np.sin(i*nj),y)))
|
|
| 39 |
Y[i]=np.sum(np.subtract(np.multiply(np.cos(i*nj),y),np.multiply(np.sin(i*nj),x)))
|
|
| 40 | 40 |
return(X,Y) |
| 41 | 41 |
|
| 42 | 42 |
import sys |
Formats disponibles : Unified diff