Skip to content
Snippets Groups Projects
Commit c56b09b4 authored by tomrink's avatar tomrink
Browse files

use meshgrid

parent 9b6f27af
No related branches found
No related tags found
No related merge requests found
...@@ -542,8 +542,8 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain ...@@ -542,8 +542,8 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain
n_x = int(xlen/w_x) n_x = int(xlen/w_x)
n_y = int(ylen/w_y) n_y = int(ylen/w_y)
# ll = [j_0 + j*w_y for j in range(n_y-1)] ll = [j_0 + j*w_y for j in range(n_y-1)]
# cc = [i_0 + i*w_x for i in range(n_x-1)] cc = [i_0 + i*w_x for i in range(n_x-1)]
for ds_name in name_list: for ds_name in name_list:
for j in range(n_y-1): for j in range(n_y-1):
...@@ -553,14 +553,14 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain ...@@ -553,14 +553,14 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain
grd_dct_n[ds_name].append(grd_dct[ds_name][j_ul:j_ul+w_y, i_ul:i_ul+w_x]) grd_dct_n[ds_name].append(grd_dct[ds_name][j_ul:j_ul+w_y, i_ul:i_ul+w_x])
# TODO: replace this with numpy meshgrid and flatten # TODO: replace this with numpy meshgrid and flatten
ll = [] # ll = []
cc = [] # cc = []
for j in range(n_y-1): # for j in range(n_y-1):
j_y = j_0 + j*w_y # j_y = j_0 + j*w_y
for i in range(n_x-1): # for i in range(n_x-1):
i_x = i_0 + i*w_x # i_x = i_0 + i*w_x
ll.append(j_y) # ll.append(j_y)
cc.append(i_x) # cc.append(i_x)
grd_dct = {name: None for name in name_list} grd_dct = {name: None for name in name_list}
for didx, ds_name in enumerate(name_list): for didx, ds_name in enumerate(name_list):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment