How can I make a for-loop pyramid more concise in Python? [duplicate]

by

By using nested for-loops you’re basically trying to create what’s known as the (Cartesian) product of the input iterables, which is what the product function, from itertools module, is for.

>>> list(product(range(3),repeat=4))
[(0, 0, 0, 0), (0, 0, 0, 1), (0, 0, 0, 2), (0, 0, 1, 0), (0, 0, 1, 1),
 (0, 0, 1, 2), (0, 0, 2, 0), (0, 0, 2, 1), (0, 0, 2, 2), (0, 1, 0, 0),
...

And in your code you can do :

for i,j,k,l in product(range(3),repeat=4):
    #do stuff

Based on python documentation, “This function is roughly equivalent to the following code, except that the actual implementation does not build up intermediate results in memory:“

def product(*args, repeat=1):
    # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
    # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
    pools = [tuple(pool) for pool in args] * repeat
    result = [[]]
    for pool in pools:
        result = [x+[y] for x in result for y in pool]
    for prod in result:
        yield tuple(prod)

More Related Contents:

  1. What are the differences between bool() and operator.truth()?
  2. Is there a reason Python 3 enumerates slower than Python 2?
  3. How to add a label to all words in a file? [closed]
  4. Why is “1000000000000000 in range(1000000000000001)” so fast in Python 3?
  5. Using both Python 2.x and Python 3.x in IPython Notebook
  6. Local variable referenced before assignment?
  7. Python Variable Declaration
  8. Integer division in Python 2 and Python 3
  9. Extracting text from a PDF file using PDFMiner in python?
  10. Why does str(float) return more digits in Python 3 than Python 2?
  11. pip or pip3 to install packages for Python 3?
  12. What’s the difference between __builtin__ and __builtins__?
  13. operator.itemgetter or lambda
  14. Most pythonic way to interleave two strings
  15. Argparse with required subparser
  16. Search for a value in a nested dictionary python
  17. portable way to write csv file in python 2 or python 3
  18. How do I run python 2 and 3 in windows 7? [duplicate]
  19. ValueError: malformed string when using ast.literal_eval
  20. Python3 correct way to import relative or absolute?
  21. Is Python’s order of evaluation of function arguments and operands deterministic (+ where is it documented)?
  22. Pip freeze vs. pip list
  23. How to run different python versions in cmd [duplicate]
  24. What is a clean way to convert a string percent to a float?
  25. Using Properties in Python classes cause “maximum recursion depth exceeded” [duplicate]
  26. Sound generation / synthesis with python?
  27. sum each value in a list of tuples
  28. Calling Python script from C++ and using its output
  29. Find the nearest point in distance for all the points in the dataset – Python
  30. What is the best approach in python: multiple OR or IN in if statement?

Leave a Comment