optimaze import

This commit is contained in:
Jianshun Gao 2023-05-19 17:36:17 +02:00
parent 8fddd8bb21
commit ab0e4cb83b

View File

@ -1,5 +1,4 @@
import xarray as xr import xarray as xr
import pandas as pd
import numpy as np import numpy as np
from datetime import datetime from datetime import datetime
@ -11,9 +10,11 @@ import finufft
def fft(dataArray, **kwargs): def fft(dataArray, **kwargs):
return xrft.fft(dataArray, **kwargs) return xrft.fft(dataArray, **kwargs)
def ifft(dataArray, **kwargs): def ifft(dataArray, **kwargs):
return xrft.ifft(dataArray, **kwargs) return xrft.ifft(dataArray, **kwargs)
def fft_nutou(dataArray, modeNum, **kwargs): def fft_nutou(dataArray, modeNum, **kwargs):
data = dataArray.to_numpy() data = dataArray.to_numpy()
@ -45,6 +46,7 @@ def fft_nutou(dataArray, modeNum, **kwargs):
return res return res
def ifft_nutou(dataArray, modeNum): def ifft_nutou(dataArray, modeNum):
pass pass