如何通过列表推导式合并两个list

如何合并两个长度相等的 list？

示例：

• 输入：

  • list1：[1, 2, 3]
  • list2: [‘a’, ‘b’, ‘c’]

• 输出：

  • [1, ‘a’, 2, ‘b’, 3, ‘c’]

方案 1：遍历两个数组

def interleave_by_loop(list1, list2):
    if len(list1) != len(list2):
        raise ValueError("传入的 list 的长度需要相等")

    new_list = []
    for index, value in enumerate(list1):
        new_list.append(value)
        new_list.append(list2[index])
    return new_list


list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']

if __name__ == "__main__":
    print(interleave_by_loop(list1, list2))
    # [1, 'a', 2, 'b', 3, 'c']
    

上面给出了一个简单的示例，就是通过一次带有索引 for 循环
每次循环中将两个 list 中对应的元素插入新的 list。

上面的方法可能是个比较通用的做法，但是在 Python 中可以写的简洁一点

方案 2

def interleave(list1, list2):
    return [val for pair in zip(list1, list2) for val in pair]


list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']

if __name__ == "__main__":
    print(interleave(list1, list2))
    # [1, 'a', 2, 'b', 3, 'c']
    

这里，我们首先观察下代码，首先

• zip(list1, list2) 把两个列表合并成一个元组列表
• 然后我们通过 for 循环得到一个个元组
• 最后把元组解析成我们的列表

其实，我们还可以修改下我们的函数，使得我们的函数更加具有通用性

def interleave(*args):
    return [val for pair in zip(*args) for val in pair]


list1 = [1, 2, 3]
list2 = ['a', 'b', 'c']
list3 = ['github', 'google.com', 'so']

if __name__ == "__main__":
    print(interleave(list1, list2, list3))
    # [1, 'a', 'github', 2, 'b', 'google.com', 3, 'c', 'so']
    

这样我们的函数就可以接受任意多个列表了！

https://github.com/fish-bugs/…