如何将numpy数组中的数据求和?
时间: 2023-06-10 15:08:34 浏览: 100
Python数据分析实践:NumPy数组常用操作new.pdf
5星 · 资源好评率100%
可以使用numpy库中的sum()函数来对数组中的数据求和。例如,以下代码将数组a中的所有数据求和并输出:
```
import numpy as np
a = np.array([1, 2, 3, 4, 5])
sum_a = np.sum(a)
print(sum_a) # 输出15
```
如果数组是多维的,可以通过指定axis参数来在指定的轴上计算和。例如,以下代码将数组b的第一个轴上的所有数据求和并输出:
```
import numpy as np
b = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
sum_b = np.sum(b, axis=0)
print(sum_b) # 输出[12 15 18]
```
阅读全文