对矩阵中每一行的元素都进行求和
时间: 2024-02-02 12:04:50 浏览: 75
求矩阵行元素之和C语言.docx
可以使用numpy库中的sum函数来对矩阵中每一行的元素都进行求和,并将结果保存在一个新的数组中。示例代码如下:
```python
import numpy as np
# 定义一个3x3的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 对每一行进行求和
row_sums = np.sum(matrix, axis=1)
print(row_sums) # 输出 [ 6 15 24]
```
在这个示例中,我们首先导入了numpy库,然后定义了一个3x3的矩阵。接着,使用`np.sum(matrix, axis=1)`的方式来对矩阵的每一行进行求和,并将结果保存在一个新的数组`row_sums`中。最后,输出结果即可。
阅读全文