矩阵删除几列代码python
时间: 2023-10-01 08:03:15 浏览: 53
python 实现 矩阵 课程设计 代码
要删除矩阵中的几列,可以使用numpy库中的delete()函数。以下是一个简单的示例代码:
```python
import numpy as np
# 创建一个3x5的矩阵
mat = np.array([[1, 2, 3, 4, 5],
[6, 7, 8, 9, 10],
[11, 12, 13, 14, 15]])
# 删除第1和第3列
new_mat = np.delete(mat, [0, 2], axis=1)
print(new_mat)
```
输出结果为:
```
array([[ 2, 4, 5],
[ 7, 9, 10],
[12, 14, 15]])
```
在上面的例子中,我们使用了delete()函数删除了矩阵中的第1和第3列,axis参数指定了删除的方向。如果axis=1,则表示按列删除,如果axis=0,则表示按行删除。
阅读全文