lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-03-15 14:50:25 +01:00
Code Issues Projects Releases Packages Wiki Activity

speed up for derived_observable when applied to 1d observables

Browse source
This commit is contained in:
Fabian Joswig 2021-10-18 10:00:15 +01:00
parent eacec6bfdc
commit e73a99409c
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pyerrors/pyerrors.py
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)