[已解决]关于两列表中数值的运算

能源系统建模

如何将两个长度相等的列表中的数值按顺序一 一对应地相加？
类似于
A=[1,2,3,4,5,6]
B=[4,5,6,7,8,9]

怎么把1和4加，2和5加，依此类推
然后得到的结果也是一个列表


求大神指导

最佳答案

月排行榜 / 总排行榜

Twilight6

2020-7-21 21:18:39

_2_ 发表于 2020-7-21 21:10
需要用到推导式

你用错了，这样即可：

A=[1,2,3,4,5,6]
B=[4,5,6,7,8,9]
print([A[i]+B[i] for i in range(len(A))])

跳转到最佳答案楼层

_2_

需要用到推导式

[i + j for i in A for j in B]

能源系统建模

_2_ 发表于 2020-7-21 21:10
需要用到推导式

这样好像是把A中的数和B中的数都两两相加了一次
我想要只加对应位置上的数，就是A中的1只跟B中的4求和

Twilight6

_2_ 发表于 2020-7-21 21:10
需要用到推导式

你用错了，这样即可：

A=[1,2,3,4,5,6]
B=[4,5,6,7,8,9]
print([A[i]+B[i] for i in range(len(A))])

_2_

能源系统建模 发表于 2020-7-21 21:18
这样好像是把A中的数和B中的数都两两相加了一次
我想要只加对应位置上的数，就是A中的1只跟B中的4求和

那也有

[A[_] + B[_] for _ in range(len(A))]

_2_

Twilight6 发表于 2020-7-21 21:18
你用错了，这样即可：

没想到你看见了……
好吧，第一次确实错了

Twilight6

_2_ 发表于 2020-7-21 21:19
那也有

你这改了和没改一样

_2_

Twilight6 发表于 2020-7-21 21:24
你这改了和没改一样

……
为什么总是想着要用两个变量呢

_2_

Twilight6 发表于 2020-7-21 21:24
你这改了和没改一样

在线丢人

BIO-张磊

原生python列表无法实现一一对应相加，可以通过for循环解决。
不过numpy库的array提供了对应加减乘除的方法
array就是数组，与list不同的是array的中的类型必须全部相同。你就记住array是numpy库处理数据的基本形式就可以了，不必深究
list可以通过numpy转化为array

import numpy as np
A=[1,2,3,4,5,6]
B=[4,5,6,7,8,9]
arrayA = np.array(A)
arrayB = np.array(B)
print(arrayA + arrayB)

list(arrayA + arrayB)

它比for循环的好处就是，numpy是经过优化的，速度会比单纯的for循环要快。

永恒的蓝色梦想

list(map(sum, zip(a, b)))

永恒的蓝色梦想

BIO-张磊 发表于 2020-7-21 21:33
原生python列表无法实现一一对应相加，可以通过for循环解决。
不过numpy库的array提供了对应加减乘除的方 ...

火箭轰蚊子。

BIO-张磊

永恒的蓝色梦想 发表于 2020-7-21 21:48
火箭轰蚊子。

怎么说呢，假如这个列表是个多维的，比如二维，这个方法就显得特别灵活了

A=[[1,2,3],[4,5,6]]
B=[[4,5,6],[7,8,9]]
np.array(A)+np.array(B)