python中让矩阵中各行同位元素的平方相加,并输出一个新矩阵
时间: 2023-09-16 14:13:01 浏览: 141
python 实现矩阵相加
5星 · 资源好评率100%
可以使用numpy库中的函数来实现这个功能。代码如下:
```python
import numpy as np
# 定义原始矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 对各行同位元素平方相加并输出新矩阵
new_matrix = np.square(matrix).sum(axis=1).reshape(-1, 1)
print(new_matrix)
```
输出结果为:
```
[[14]
[77]
[194]]
```
这是每一行元素平方相加的结果组成的矩阵。
阅读全文