Converting a loop with an assignment into a comprehension [duplicate]

by

You can’t directly translate that loop to a comprehension. Comprehensions, being expressions, can only contain expressions, and assignments are statements.

However, that doesn’t mean there are no options.

First, at the cost of calling split twice, you can just do this:

mylist = [word.split('l')[0]+word.split('l')[1] for word in ['Hello', 'world']]

But you don’t want to call split twice.

The most general way around that is to use a chain of generator expressions (with one list comprehension at the end) to transform things:

words = (word.split('l') for word in ['Hello', 'world'])
mylist = [w[0]+w[1] for w in words]

If you really want to merge that all into one expression, you can:

mylist = [w[0]+w[1] for w in 
          (word.split('l') for word in ['Hello', 'world'])]

But unless you actually need it to be in an expression, it’s probably more readable not to do that.

A more specific way in this case is to replace the w[0]+w[1] with something equivalent that doesn’t need to reference w twice:

mylist = [''.join(word.split('l')[:2]) for word in ['Hello', 'world']]

And you can always generalize this one, too. You can turn any expression into a function, which means you can avoid evaluating any part of it by passing it as an argument to that function. If there isn’t a function that does what you want, write it:

def join_up(split_word):
    return split_word[0]+split_word[1]
mylist = [join_up(word.split('l')) for word in ['Hello', 'world']]

If you need to make that all into one expression without repeating any work, it may not be pretty:

mylist = [(lambda split_word: split_word[0]+split_word[1])(word.split('l')) 
          for word in ['Hello', 'world']]

But ultimately, unless I already had a function lying around that did what I needed, I’d use the chain-of-generator-expressions solution.

Or, of course, just keep it in an explicit loop; there’s nothing wrong with for loops, and if the intermediate temporary variable makes your code clearer, there’s no better way to do that than with an assignment statement.

More Related Contents:

  1. Add Text on Image using PIL
  2. Concatenate two PySpark dataframes
  3. Line continuation for list comprehensions or generator expressions in python
  4. OpenCV & Python – Image too big to display
  5. How to install PyQt5 on Windows?
  6. conda environment in google colab [google-colaboratory]
  7. How can I abort a task in a multiprocessing.Pool after a timeout?
  8. How to use pyinstaller?
  9. How can i parse a comma delimited string into a list (caveat)?
  10. How to remove small connected objects using OpenCV
  11. Check if list contains only item x
  12. Using SQLAlchemy models in and out of Flask
  13. How to override template in django-allauth?
  14. Flask, SQLAlchemy : KeyError: ‘SQLALCHEMY_TRACK_MODIFICATIONS’
  15. Plotting errors bars from dataframe using Seaborn FacetGrid
  16. Reading a File Line by Line in Python [duplicate]
  17. How to combine year, month, and day columns to single datetime column?
  18. using variable in import command
  19. What is the most efficient way of doing square root of sum of square of two numbers?
  20. Merge dataframes in a dictionary
  21. How to understand Seaborn’s heatmap annotation format
  22. Importing module from string variable using “__import__” gives different results than a normal import statement
  23. Convert bytes to a string in python 3
  24. How to run IPython magic from a script
  25. FastAPI python: How to run a thread in the background?
  26. Scatter plot with legend for each color in c
  27. Python strip() multiple characters?
  28. How do I import a Python script from a sibling directory?
  29. When will Jython support Python 3? [closed]
  30. Python3 error: initial_value must be str or None, with StringIO

Leave a Comment