Skip to content

ranges

Functions:

interleave_arr 

interleave_arr(
    arr0: Iterable[T0], arr1: Iterable[T1], n: int = 2
) -> Iterable[T0 | T1]

Interleave two arrays of variable length.

Parameters:

  • arr0

    (Iterable[T0]) –

    First array to be interleaved.

  • arr1

    (Iterable[T1]) –

    Second array to be interleaved.

  • n

    (int, default: 2 ) –

    The number of elements from arr0 to include in the interleaved sequence before including an element from arr1.

Yields: Elements from either arr0 or arr01.

Source code in jetpytools/utils/ranges.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
def interleave_arr[T0, T1](arr0: Iterable[T0], arr1: Iterable[T1], n: int = 2) -> Iterable[T0 | T1]:
    """
    Interleave two arrays of variable length.

    Args:
        arr0: First array to be interleaved.
        arr1: Second array to be interleaved.
        n: The number of elements from arr0 to include in the interleaved sequence before including an element from
            arr1.
    Yields:
        Elements from either arr0 or arr01.
    """

    if n <= 0:
        raise ValueError("n must be a positive integer")

    if n == 1:
        for x in chain.from_iterable(zip_longest(arr0, arr1, fillvalue=None)):
            if x is not None:
                yield x
        return

    arr0_i, arr1_i = iter(arr0), iter(arr1)
    arr1_vals = arr0_vals = True

    while arr1_vals or arr0_vals:
        if arr0_vals:
            for _ in range(n):
                try:
                    yield next(arr0_i)
                except StopIteration:
                    arr0_vals = False

        if arr1_vals:
            try:
                yield next(arr1_i)
            except StopIteration:
                arr1_vals = False