I thought that itertools.chain.from_iterable isn't useful.
cuz this only allow "single-nested iterable -- this will raise error when 
arg has non-nested element--" like below::
>>> from itertools import chain
>>> chain.from_iterable([1])
<itertools.chain object at 0x00000112D332BF98>
>>> list(_)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'int' object is not iterable

and this can't unpack over double nest.
>>> chain.from_iterable([[[1, 2]], [[4, 5]]])
<itertools.chain object at 0x00000112D3330B00>
>>> list(_)
[[1, 2], [4, 5]]
 
So, I wanted to make "True chain.from_iterable". and this is it.

def flatten(iterables, unpack=(list, tuple, set), peep=(list, tuple, set)):
    for element in iterables:
        try:
            if isinstance(element, unpack):
                if isinstance(element, peep):
                    yield from flatten(element, unpack=unpack, peep=peep)
                else:
                    yield from flatten(element, unpack=(), peep=())
            elif isinstance(element, peep):
                yield type(element)(flatten(element, unpack=unpack, peep=peep))
            else:
                raise TypeError
        except TypeError:
            yield element


Reason why I didin't use type() is wanted to unpack type-object like "range"
and I wanted to unpack user-defined-class/func. this is why I didin't use 
collections.Iterable to check instance.

I know this will be destructed by itertools.count :(

Please give me advice.

And I wanna know why function like this is not in standard library.

thx.
_______________________________________________
Python-ideas mailing list
Python-ideas@python.org
https://mail.python.org/mailman/listinfo/python-ideas
Code of Conduct: http://python.org/psf/codeofconduct/

Reply via email to