python两个列表相加

本文讨论两个列表如何相加在一起,要求这两个列表满足以下要求

  1. 列表长度相同
  2. 列表里的元素都是int类型数据
  3. 对应索引位置元素相加,生成新的列表

示例如下

输入:
lst1 = [1, 4, 7]
lst2 = [2, 5, 3]

输出:
sum_lst = [3, 9, 10]

方法1, 遍历相加

lst1 = [1, 4, 7]
lst2 = [2, 5, 3]

sum_lst = []
for index, item in enumerate(lst1):
    sum_lst.append(item + lst2[index])

print(sum_lst)

能想到这样的算法,就算是入门合格了,掌握了python基础技术,懂得如何遍历列表,如何向列表里新增元素

方法2, 使用列表推导式

使用列表推导式,并结合zip函数

lst1 = [1, 4, 7]
lst2 = [2, 5, 3]

sum_lst = [x + y for x, y in zip(lst1, lst2)]
print(sum_lst)

方法2的核心在于正确使用zip函数,zip函数将多个列表视为一个整体,遍历时,同时从各个列表里取出相同索引位置的元素

方法3, 使用map

lst1 = [1, 4, 7]
lst2 = [2, 5, 3]

import operator
sum_lst = list(map(operator.add, lst1, lst2))
print(sum_lst)

map函数从lst1和lst2中各取出一个数据然后使用operator.add相加, operator.add也可以替换成lambda表达式

lst1 = [1, 4, 7]
lst2 = [2, 5, 3]

sum_lst = list(map(lambda x, y: x + y, lst1, lst2))
print(sum_lst)

方法4, 使用numpy

lst1 = [1, 4, 7]
lst2 = [2, 5, 3]

import numpy as np
sum_lst = list(np.add(lst1, lst2))
print(sum_lst)

扫描关注, 与我技术互动

QQ交流群: 211426309

加入知识星球, 每天收获更多精彩内容

分享日常研究的python技术和遇到的问题及解决方案