Skip to content
Snippets Groups Projects
Commit c38a363d authored by Alán Muñoz's avatar Alán Muñoz
Browse files

feat(cells): add at_times method

parent 9ec11cc5
No related branches found
No related tags found
No related merge requests found
...@@ -162,6 +162,15 @@ class Cells: ...@@ -162,6 +162,15 @@ class Cells:
] ]
return self.group_by_traps(traps, masks) return self.group_by_traps(traps, masks)
def at_times(self, timepoints: t.Iterable[int], kind="mask"):
return [
[
np.dstack(tile_masks) if len(tile_masks) else []
for tile_masks in self.at_time(tp, kind=kind).values()
]
for tp in timepoints
]
def group_by_traps( def group_by_traps(
self, traps: t.Collection, cell_labels: t.Collection self, traps: t.Collection, cell_labels: t.Collection
) -> t.Dict[int, t.List[int]]: ) -> t.Dict[int, t.List[int]]:
......
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