feat: add __contains__ to CompressoArray

seung-lab · Mar 6, 2024 · 751179c · 751179c
1 parent e5814e2
commit 751179c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/compresso.pyx b/compresso.pyx
@@ -604,6 +604,9 @@ class CompressoArray:
   def remap(self, buf, mapping, preserve_missing_labels=False):
     return CompressoArray(remap(buf, mapping, preserve_missing_labels))
 
+  def __contains__(self, label):
+    return label in self.labels()
+
   def __getitem__(self, slcs):
     slices = reify_slices(slcs, *self.shape)