python怎么删除对角线,把3✖️3矩阵变成3✖️2矩阵
时间: 2024-03-18 20:40:07 浏览: 46
Python二维数组实现求出3*3矩阵对角线元素的和示例
5星 · 资源好评率100%
可以使用 numpy 库中的 delete 函数来删除对角线,然后重新构造一个新的3x2矩阵。
示例代码如下:
```python
import numpy as np
# 定义一个3x3矩阵
mat = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 删除对角线上的元素
new_mat = np.delete(mat, [0, 4, 8])
# 重新构造一个3x2矩阵
new_mat = new_mat.reshape(3, 2)
print(new_mat)
```
输出结果为:
```
[[2 3]
[4 6]
[7 8]]
```
在上述代码中,我们首先定义了一个3x3的矩阵 mat,然后使用 np.delete 函数删除了对角线上的元素,得到了一个新的一维数组 new_mat。接着,我们使用 reshape 函数将 new_mat 重新构造成一个3x2的矩阵,并打印出来。
阅读全文