bug in derived_array for irregular monte carlo chains fixed

This commit is contained in:
Fabian Joswig 2021-11-11 11:07:15 +00:00
parent b5503cb362
commit 7c0ff6c8d9

View file

@ -86,7 +86,7 @@ def derived_array(func, data, **kwargs):
for name in new_names:
d_extracted[name] = []
for i_dat, dat in enumerate(data):
ens_length = dat.ravel()[0].shape[name]
ens_length = new_idl_d[name][-1] - new_idl_d[name][0] + 1
d_extracted[name].append(np.array([_expand_deltas_for_merge(o.deltas[name], o.idl[name], o.shape[name], new_idl_d[name]) for o in dat.reshape(np.prod(dat.shape))]).reshape(dat.shape + (ens_length, )))
for i_val, new_val in np.ndenumerate(new_values):