sum对一行多列的矩阵求和
时间: 2024-05-07 11:17:35 浏览: 68
您可以使用numpy库中的sum函数来实现这个功能。具体实现如下:
```python
import numpy as np
# 定义一个3行4列的矩阵
matrix = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
# 对每一行求和
row_sum = matrix.sum(axis=1)
print(row_sum)
# 输出结果为 [10 26 42]
# 对每一列求和
col_sum = matrix.sum(axis=0)
print(col_sum)
# 输出结果为 [15 18 21 24]
```
在上面的例子中,axis=1表示对每一行进行求和,而axis=0表示对每一列进行求和。
阅读全文