Source code for ibllib.misc.flatten

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Author: Niccolò Bonacchi
# @Date: Wednesday, July 25th 2018, 3:56:27 pm
Flatten a nested Iterable excluding strings and dicts.

Converts nested Iterable into flat list. Will not iterate through strings or

:return: Flattened list or generator object.
:rtype: list or generator
import collections

[docs]def iflatten(x): result = [] for el in x: if isinstance(el, and not ( isinstance(el, str) or isinstance(el, dict)): result.extend(iflatten(el)) else: result.append(el) return result
[docs]def gflatten(x): def iselement(e): return not(isinstance(e, and not( isinstance(el, str) or isinstance(el, dict))) for el in x: if iselement(el): yield el else: yield from gflatten(el)
[docs]def flatten(x, generator=False): if generator: return gflatten(x) return iflatten(x)