Function flatten_depth

botasaurus/list_utils.py:10–34

Recursively flatten `array` up to `depth` times. Args: array (list): List to flatten. depth (int, optional): Depth to flatten to. Defaults to ``1``. Returns: list: Flattened list. Example: >>> flatten_depth([[[1], [2, [3]], [[4]]]], 1)

(array, depth=1)

            yield item

def flatten_depth(array, depth=1):
    """
    Recursively flatten `array` up to `depth` times.

    Args:
        array (list): List to flatten.
        depth (int, optional): Depth to flatten to. Defaults to ``1``.

    Returns:
        list: Flattened list.

    Example:

        >>> flatten_depth([[[1], [2, [3]], [[4]]]], 1)
        [[1], [2, [3]], [[4]]]
        >>> flatten_depth([[[1], [2, [3]], [[4]]]], 2)
        [1, 2, [3], [4]]
        >>> flatten_depth([[[1], [2, [3]], [[4]]]], 3)
        [1, 2, 3, 4]
        >>> flatten_depth([[[1], [2, [3]], [[4]]]], 4)
        [1, 2, 3, 4]

    .. versionadded:: 4.0.0
    """
    return list(iterflatten(array, depth=depth))

def flatten(array):
    """