From 9e5f1d17a4b0165bd4a3218ca24a159012ea49bd Mon Sep 17 00:00:00 2001 From: Jonas Gunz Date: Mon, 14 Oct 2024 00:55:26 +0200 Subject: misc --- misc.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'misc.py') diff --git a/misc.py b/misc.py index dcbd2f8..6594d0f 100644 --- a/misc.py +++ b/misc.py @@ -2,12 +2,18 @@ import os import numpy as np import datetime -def np_time_convert(dt64): +def np_time_convert(dt64, func=datetime.datetime.utcfromtimestamp): unix_epoch = np.datetime64(0, 's') one_second = np.timedelta64(1, 's') seconds_since_epoch = (dt64 - unix_epoch) / one_second - return datetime.datetime.utcfromtimestamp(seconds_since_epoch) + return func(seconds_since_epoch) + +def np_time_convert_offset(init, step): + return np_time_convert(init) + np_time_convert(step, func=lambda x: datetime.timedelta(seconds=x)) + +def np_time_list_convert_offset(init, steps): + return list(map(lambda x: np_time_convert_offset(init, x), steps)) def create_output_dir(path, clear=False): if not os.path.exists(path): -- cgit v1.2.3