Skip to content

GitLab

Commit 46cb64ba authored by tpys's avatar tpys
Browse files

move save_like to data_util

parent c56e400c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
data_util.py
View file @ 46cb64ba
...@@ -30,7 +30,6 @@ def save_like(output, input, step, save_dir="", freq=6): ...@@ -30,7 +30,6 @@ def save_like(output, input, step, save_dir="", freq=6):
os.makedirs(save_dir, exist_ok=True) os.makedirs(save_dir, exist_ok=True)
dtime = (step+2) * freq # dtime = (step+2) * freq #
init_time = pd.to_datetime(input.time.values[0]) init_time = pd.to_datetime(input.time.values[0])
print(f'init_time: {init_time}, dtime: {dtime}')
data = xr.DataArray( data = xr.DataArray(
output[None, None], output[None, None],
...@@ -60,6 +59,7 @@ def save_like(output, input, step, save_dir="", freq=6): ...@@ -60,6 +59,7 @@ def save_like(output, input, step, save_dir="", freq=6):
ds.append(v) ds.append(v)
ds = xr.merge(ds, compat="no_conflicts") ds = xr.merge(ds, compat="no_conflicts")
print(f'Save to {save_name} ...')
save_name = os.path.join(save_dir, f'{dtime:03d}.nc') save_name = os.path.join(save_dir, f'{dtime:03d}.nc')
ds.to_netcdf(save_name) ds.to_netcdf(save_name)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment