[toyg]

I've not tried, but you can probably get that by recursively mixing standard constructs and functools.chain.from_iterable().

[OskarS]

You should try. It's harder than it seems.

(I mean, it's not the most challenging problem ever, but most programmers look at it and go "that's trivial, just do X!", and it's a bit trickier than that).