python元素明智地与其他列表一起添加到列表列表中

Question

我在 sum of N lists element-wise python 检查了类似的问题

但我的有点复杂，因为我正在添加列表列表和另一个列表列表

我的列表如下所示

[[0         0.014285714 0.035600016]
,[0.014285714   0           0.038359389]
,[0.035600016   0.038359389 0]]


[[0         0.014285714 0.035600016]
,[0.014285714   0           0.038359389]
,[0.035600016   0.038359389 0]]

所以将这两个相加然后结果有 3 x 3 矩阵

我怎样才能有效地添加这两个列表？

Answer 1

我可以用 numpy 解决吗:

>>> a = [[1, 2, 3], [1, 2, 3], [1, 2, 3]]
>>> b = [[4, 5, 6], [4, 5, 6], [4, 5, 6]]

>>> from numpy import array, sum

>>> list(array(a) + array(b))
[array([5, 7, 9]), array([5, 7, 9]), array([5, 7, 9])]

或

>>> sum([a, b], axis=0)
 array([[5, 7, 9],
       [5, 7, 9],
       [5, 7, 9]])
>>>