Python sum() 函数的一点用法
- 2020-09-15 10:09:00
- 六月
- 来源:
- https://fishc.com.cn/forum.php?mod=viewthread&tid=84912&ctid=588
- 转贴 848
两个 3 行 3 列的矩阵,实现其对应位置的数据相加,并返回一个新矩阵:
X = [[12,7,3],
[4,5,6],
[7,8,9]]
Y = [[5,8,1],
[6,7,3],
[4,5,9]]
程序分析:创建一个新的 3 行 3 列的矩阵,使用 for 迭代并取出 X 和 Y 矩阵中对应位置的值,相加后放到新矩阵的对应位置中。
对应的实现代码:
x = [[12,7,3],
[4,5,6],
[7,8,9]]
y = [[5,8,1],
[6,7,3],
[4,5,9]]
z = [i + j for i, j in zip(sum(x, []), sum (y, []))]
z = [z[0:3], z[3:6], z[6:9]]
这里红色字体标准部分涉及到zip函数,以及sum函数的用法,常见的我们sum函数常用参数是迭代对象:列表(list)、元组(tuple)、集合(set)、字典(dictionary)。这里我们看到sum函数对两个列表进行了相加,其中x是元素为列表的列表,结果得到的是一个整形元素列表。
sum(x, [])结果是:[12, 7, 3, 4, 5, 6, 7, 8, 9]
同理,我们可以使用sum对元组,字典,集合进行类似的操作。