Python 3 replacement for deprecated compiler.ast flatten function

by

itertools.chain is the best solution for flattening any nested iterable one level – it’s highly efficient compared to any pure-python solution.

That said, it will work on all iterables, so some checking is required if you want to avoid it flattening out strings, for example.

Likewise, it won’t magically flatten out to an arbitrary depth. That said, generally, such a generic solution isn’t required – instead it’s best to keep your data structured so that it doesn’t require flattening in that way.

Edit: I would argue that if one had to do arbitrary flattening, this is the best way:

import collections

def flatten(iterable):
    for el in iterable:
        if isinstance(el, collections.Iterable) and not isinstance(el, str): 
            yield from flatten(el)
        else:
            yield el

Remember to use basestring in 2.x over str, and for subel in flatten(el): yield el instead of yield from flatten(el) pre-3.3.

As noted in the comments, I would argue this is the nuclear option, and is likely to cause more problems than it solves. Instead, the best idea is to make your output more regular (output that contains one item still give it as a one item tuple, for example), and do regular flattening by one level where it is introduced, rather than all at the end.

This will produce more logical, readable, and easier to work with code. Naturally, there are cases where you need to do this kind of flattening (if the data is coming from somewhere you can’t mess with, so you have no option but to take it in the poorly-structured format), in which case, this kind of solution might be needed, but in general, it’s probably a bad idea.

More Related Contents:

  1. how to unpack a tuple in this form (‘abc’, [20, 20])? [closed]
  2. How to pass arguments to a Button command in Tkinter?
  3. Django MEDIA_URL and MEDIA_ROOT
  4. TypeError: a bytes-like object is required, not ‘str’
  5. pandas: merge (join) two data frames on multiple columns
  6. Pandas Extract Number from String
  7. Could not load dynamic library ‘cudart64_101.dll’ on tensorflow CPU-only installation
  8. Is it possible to change sprite colours in Pygame?
  9. asyncio.sleep() vs time.sleep()
  10. How to flatten a pandas dataframe with some columns as json?
  11. Removing the TK icon on a Tkinter window
  12. Concurrent.futures vs Multiprocessing in Python 3
  13. Installing Python 3 on RHEL
  14. Python TypeError: non-empty format string passed to object.__format__
  15. Count number of words per row
  16. Tkinter: How to get frame in canvas window to expand to the size of the canvas?
  17. How do I use a PyGame timer event? How to add a clock to a pygame screen using a timer?
  18. cleanest way to call one function on a list of items
  19. Get folder name of the file in Python
  20. pythoncom crashes on KeyDown when used hooked to certain applications
  21. How PyCharm imports differently than system command prompt (Windows)
  22. time.time vs. timeit.timeit
  23. How do the scoping rules work with classes?
  24. Leading zeros are not allowed in Python?
  25. Can’t import annotations from __future__
  26. How to use sys.exit() in Python
  27. Is there a way to use asyncio.Queue in multiple threads?
  28. How to get value of variable entered from user input?
  29. Add custom form fields that are not part of the model (Django)
  30. Writing to a new file if it doesn’t exist, and appending to a file if it does

Leave a Comment