使用np.sum在python中添加Numpy数组?

时间:2018-10-30 00:12:21

标签: python arrays numpy

标题基本上可以解释这一点,我是数组的新手,我需要添加列,这是我的数组:

[[20,40,60,21,70]
[2,76,5,90,50]
[21,67,99,33,49]]

所以我需要添加(20 + 2 + 21),(40 + 76 + 67)等。

我尝试使用np.sum,但是我对数组有些陌生,所以我不确定如何正确地对数组进行切片/索引。

谢谢您的帮助

1 个答案:

答案 0 :(得分:2)

在求和函数中使用axis=0

a = np.arange(15).reshape(-1,5)
# array([[ 0,  1,  2,  3,  4],
#   [ 5,  6,  7,  8,  9],
#   [10, 11, 12, 13, 14]])
np.sum(a, axis=0)
# array([15, 18, 21, 24, 27])