numpy中的ndarray函数如何对各行求和的例子
时间: 2023-05-23 19:00:57 浏览: 83
可以使用numpy中的sum函数对ndarray数组的各行求和,例如:
``` python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 对各行求和
sums = np.sum(a, axis=1)
print(sums)
# 输出:[ 6 15 24]
```
以上代码中,np.sum(a, axis=1)表示对数组a中的各行求和,axis=1表示在第二个维度上(即列)进行求和。输出结果为每一行的和,即[6 15 24]。
阅读全文