Skip to content
Snippets Groups Projects

Resolve "release v1.4.0"

Merged Ghost User requested to merge release_v1.4.0 into master
2 files
+ 8
4
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -17,7 +17,7 @@ from matplotlib import lines as mlines, pyplot as plt, patches as mpatches, date
@@ -17,7 +17,7 @@ from matplotlib import lines as mlines, pyplot as plt, patches as mpatches, date
from astropy.timeseries import LombScargle
from astropy.timeseries import LombScargle
from mlair.data_handler import DataCollection
from mlair.data_handler import DataCollection
from mlair.helpers import TimeTrackingWrapper, to_list
from mlair.helpers import TimeTrackingWrapper, to_list, remove_items
from mlair.plotting.abstract_plot_class import AbstractPlotClass
from mlair.plotting.abstract_plot_class import AbstractPlotClass
@@ -829,9 +829,13 @@ class PlotPeriodogram(AbstractPlotClass): # pragma: no cover
@@ -829,9 +829,13 @@ class PlotPeriodogram(AbstractPlotClass): # pragma: no cover
plt.close('all')
plt.close('all')
def f_proc(var, d_var, f_index): # pragma: no cover
def f_proc(var, d_var, f_index, time_dim="datetime"): # pragma: no cover
var_str = str(var)
var_str = str(var)
t = (d_var.datetime - d_var.datetime[0]).astype("timedelta64[h]").values / np.timedelta64(1, "D")
t = (d_var[time_dim] - d_var[time_dim][0]).astype("timedelta64[h]").values / np.timedelta64(1, "D")
 
if len(d_var.shape) > 1: # use only max value if dimensions are remaining (e.g. max(window) -> latest value)
 
to_remove = remove_items(d_var.coords.dims, time_dim)
 
for e in to_list(to_remove):
 
d_var = d_var.sel({e: d_var[e].max()})
pgram = LombScargle(t, d_var.values.flatten(), nterms=1, normalization="psd").power(f_index)
pgram = LombScargle(t, d_var.values.flatten(), nterms=1, normalization="psd").power(f_index)
# f, pgram = LombScargle(t, d_var.values.flatten(), nterms=1, normalization="psd").autopower()
# f, pgram = LombScargle(t, d_var.values.flatten(), nterms=1, normalization="psd").autopower()
return var_str, f_index, pgram
return var_str, f_index, pgram
Loading