Zipping lists of unequal size

by

Normally, you use itertools.zip_longest for this:

>>> import itertools
>>> a = [1, 2, 3]
>>> b = [9, 10]
>>> for i in itertools.zip_longest(a, b): print(i)
... 
(1, 9)
(2, 10)
(3, None)

But zip_longest pads the shorter iterable with Nones (or whatever value you pass as the fillvalue= parameter). If that’s not what you want then you can use a comprehension to filter out the Nones:

>>> for i in (tuple(p for p in pair if p is not None) 
...           for pair in itertools.zip_longest(a, b)):
...     print(i)
... 
(1, 9)
(2, 10)
(3,)

but note that if either of the iterables has None values, this will filter them out too. If you don’t want that, define your own object for fillvalue= and filter that instead of None:

sentinel = object()

def zip_longest_no_fill(a, b):
    for i in itertools.zip_longest(a, b, fillvalue=sentinel):
        yield tuple(x for x in i if x is not sentinel)

list(zip_longest_no_fill(a, b))  # [(1, 9), (2, 10), (3,)]

More Related Contents:

  1. What is the difference between Python’s list methods append and extend?
  2. How do I sort a list of dictionaries by a value of the dictionary?
  3. What is the best way to implement nested dictionaries?
  4. How are Python’s Built In Dictionaries Implemented?
  5. Python Sets vs Lists
  6. What would a “frozen dict” be?
  7. In Python, when to use a Dictionary, List or Set?
  8. How to convert SQL Query result to PANDAS Data Structure?
  9. How can I implement a tree in Python?
  10. Hashable, immutable
  11. How to implement a binary tree?
  12. Time complexity of python set operations?
  13. What do I use for a max-heap implementation in Python?
  14. How can I extract a single value from a JSON response?
  15. Removing elements from a list containing specific characters [duplicate]
  16. List of all unique characters in a string?
  17. Representing graphs (data structure) in Python
  18. What is the purpose of collections.ChainMap?
  19. Key-ordered dict in Python
  20. Creating a tree/deeply nested dict from an indented text file in python
  21. Using a Python Dictionary as a Key (Non-nested)
  22. How to implement a binary search tree in Python?
  23. Recursive diff of two dictionaries (keys and values)?
  24. How to make a dictionary that returns key for keys missing from the dictionary instead of raising KeyError?
  25. How can I extract a single value from a nested data structure (such as from parsing JSON)?
  26. How to get ordered dictionaries in pymongo?
  27. How expensive are Python dictionaries to handle?
  28. Adding counters deletes keys
  29. How to make JSON flattening memory efficient?
  30. How to sort a list of dictionaries by a value of the dictionary in Python?

Leave a Comment