hiro/hopsflow
1
0
Fork 0
mirror of https://github.com/vale981/hopsflow synced 2025-03-05 08:51:37 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

add slice to ensemble value

Browse source
This commit is contained in:
Valentin Boettcher 2022-07-22 17:03:29 +02:00
parent eee8cbedef
commit 913dde147d
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
hopsflow/util.py
View file

@ -115,6 +115,15 @@ class EnsembleValue:
def __getitem__(self, index): def __getitem__(self, index):
return EnsembleValue(self._value[index]) return EnsembleValue(self._value[index])
def slice(self, *args):
slc = slice(*args)
results = []
for N, val, σ in self.aggregate_iterator:
results.append((N, val[slc], σ[slc]))
return EnsembleValue(results)
def __len__(self) -> int: def __len__(self) -> int:
return len(self._value) return len(self._value)