merge
Python, List

Merges two or more lists into a list of lists, combining elements from each of the input lists based on their positions.

def merge(*args, fill_value = None):
  max_length = max([len(lst) for lst in args])
  result = []
  for i in range(max_length):
    result.append([
      args[k][i] if i < len(args[k]) else fill_value for k in range(len(args))
    ])
  return result

Python Codes

merge
Python, List

Merges two or more lists into a list of lists, combining elements from each of the input lists based on their positions.

.👉🏻 Use max() combined with a list comprehension to get the length of the longest list in the arguments.

.👉🏻 Use range() in combination with the max_length variable to loop as many times as there are elements in the longest list.

.👉🏻If a list is shorter than max_length, use fill_value for the remaining items (defaults to None).

.👉🏻zip() and itertools.zip_longest() provide similar functionality to this snippet.


Code:

def merge(*args, fill_value = None):
  max_length = max([len(lst) for lst in args])
  result = []
  for i in range(max_length):
    result.append([
      args[k][i] if i < len(args[k]) else fill_value for k in range(len(args))
    ])
  return result

Example:

merge(['a', 'b'], [1, 2], [True, False])

Output: [['a', 1, True], ['b', 2, False]]

Share and Support
@Python_Codes

hottg.com/python_codes/161

13.1K viewsFeb 16, 2021 at 15:46