Python代码阅读(第17篇):列表元素按位合并

2021年6月21日 3点热度 0条评论 来源: FelixZ

Python 代码阅读合集介绍:为什么不推荐Python初学者直接看项目源码

本篇阅读的代码实现了多个列表按照元素的位置依序合并成一个新的嵌套列表。每个子列表都是原列表对应位置的元素。zip()itertools.zip_longest()提供了类似于这段代码的功能。

本篇阅读的代码片段来自于30-seconds-of-python

merge

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

# EXAMPLES
merge(['a', 'b'], [1, 2], [True, False]) # [['a', 1, True], ['b', 2, False]]
merge(['a'], [1, 2], [True, False]) # [['a', 1, True], [None, 2, False]]
merge(['a'], [1, 2], [True, False], fill_value = '_') # [['a', 1, True], ['_', 2, False]]

merge函数接收多个列表和一个填充值fill_value,返回一个嵌套列表。该嵌套列表的元素个数等于最长的输入列表的长度,每个子列表的元素依次来自于输入列表的相同位置,输入列表的元素不足的情况下,使用填充值代替。也就是说merge_list[0]是一个列表,其中的元素依次为输入列表索引为0的元素。

函数使用max结合list comprehension得到参数中最长的列表的长度。结合使用range()max_length变量来循环。如果一个列表的长度短于max_length,则剩余的项目使用fill_value(默认为None)来进行填充。

函数调用同时使用了“关键字参数”和“可变参数”的方式。在第9篇的时候介绍了单独使用“可变参数”调用的方式。一般来说,这些“可变参数”将在形式参数列表的末尾,因为它们收集传递给函数的所有剩余输入参数。出现在*args参数之后的任何形式参数都是“仅限关键字参数”。

    原文作者:FelixZ
    原文地址: https://www.cnblogs.com/felixz/p/15420425.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系管理员进行删除。