speed up for derived_observable when applied to 1d observables

This commit is contained in:
Fabian Joswig 2021-10-18 10:00:15 +01:00
parent eacec6bfdc
commit e73a99409c

View file

@ -758,8 +758,10 @@ def derived_observable(func, data, **kwargs):
else:
if new_shape[name] != tmp:
raise Exception('Shapes of ensemble', name, 'do not match.')
values = np.vectorize(lambda x: x.value)(data)
if data.ndim == 1:
values = np.array([o.value for o in data])
else:
values = np.vectorize(lambda x: x.value)(data)
new_values = func(values, **kwargs)