ethograph.labels.ml.find_blocks#

ethograph.labels.ml.find_blocks(mask)[source]#

Find contiguous True blocks in a boolean array.

Parameters:

mask (np.ndarray) – Boolean array.

Return type:

tuple[ndarray, ndarray]

Returns:

  • starts (np.ndarray) – Start indices of True blocks.

  • ends (np.ndarray) – End indices (inclusive) of True blocks.

Examples

>>> import numpy as np
>>> mask = np.array([False, True, True, False, True])
>>> starts, ends = find_blocks(mask)
>>> starts
array([1, 4])
>>> ends
array([2, 4])