neuralib.util.segments.grouped_iter

neuralib.util.segments.grouped_iter(v, n)[source]

Groups elements from the input iterable v into tuples of length n

>>> list(grouped_iter([1, 2, 3, 4, 5, 6], 2))
[(1, 2), (3, 4), (5, 6)]
Parameters:

  • v (ndarray[tuple[Any, ...], dtype[Any]] | Sequence[Any] | pd.Series | pl.Series | Iterable) – input iterable to be grouped.

  • n (int) – number of elements per group

Returns:

An iterator over tuples of length n. Trailing elements are dropped if len(v) is not a multiple of n.

Return type:

zip