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

minor...

parent 171d5d00
No related branches found
No related tags found
No related merge requests found
...@@ -444,6 +444,10 @@ def process(ice_dct, no_ice_dct, neg_ice_dct): ...@@ -444,6 +444,10 @@ def process(ice_dct, no_ice_dct, neg_ice_dct):
ice_keys = np.concatenate([ice_keys_5_6, ice_keys_1, ice_keys_2, ice_keys_3, ice_keys_4]) ice_keys = np.concatenate([ice_keys_5_6, ice_keys_1, ice_keys_2, ice_keys_3, ice_keys_4])
ice_tidx = np.concatenate([ice_tidx_5_6, ice_tidx_1, ice_tidx_2, ice_tidx_3, ice_tidx_4]) ice_tidx = np.concatenate([ice_tidx_5_6, ice_tidx_1, ice_tidx_2, ice_tidx_3, ice_tidx_4])
sidxs = np.argsort(ice_keys)
ice_keys = ice_keys[sidxs]
ice_tidx = ice_tidx[sidxs]
for idx, key in enumerate(ice_keys): for idx, key in enumerate(ice_keys):
rpts = ice_dct[key] rpts = ice_dct[key]
tup = rpts[ice_tidx[idx]] tup = rpts[ice_tidx[idx]]
...@@ -466,6 +470,10 @@ def process(ice_dct, no_ice_dct, neg_ice_dct): ...@@ -466,6 +470,10 @@ def process(ice_dct, no_ice_dct, neg_ice_dct):
no_ice_tidx = no_ice_tidx[::20] no_ice_tidx = no_ice_tidx[::20]
print('no ice : ', no_ice_keys.shape[0]) print('no ice : ', no_ice_keys.shape[0])
sidxs = np.argsort(no_ice_keys)
no_ice_keys = no_ice_keys[sidxs]
no_ice_tidx = no_ice_tidx[sidxs]
for idx, key in enumerate(no_ice_keys): for idx, key in enumerate(no_ice_keys):
rpts = no_ice_dct[key] rpts = no_ice_dct[key]
tup = rpts[no_ice_tidx[idx]] tup = rpts[no_ice_tidx[idx]]
......
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