numpy 矩阵中间插入3列
时间: 2023-08-11 15:55:20 浏览: 186
要在NumPy矩阵的中间插入3列,你可以使用NumPy的`hstack()`函数和切片操作来实现。以下是一个示例代码:
```python
import numpy as np
# 创建一个3行5列的矩阵
matrix = np.array([[1, 2, 3, 4, 5],
[6, 7, 8, 9, 10],
[11, 12, 13, 14, 15]])
# 在第2列后插入3列
insert_cols = np.array([[100, 200, 300],
[400, 500, 600],
[700, 800, 900]])
# 使用hstack函数将矩阵拆分为左右两部分,并在中间插入新的列
left_part = matrix[:, :2]
right_part = matrix[:, 2:]
# 将左部分、插入的列和右部分重新拼接为一个新的矩阵
new_matrix = np.hstack((left_part, insert_cols, right_part))
print(new_matrix)
```
这将输出以下结果:
```
[[ 1 2 100 200 300 3 4 5]
[ 6 7 400 500 600 8 9 10]
[ 11 12 700 800 900 13 14 15]]
```
在这个示例中,我们首先创建了一个3行5列的矩阵。然后,我们定义了要插入的3列数据。接下来,我们使用切片操作将原始矩阵拆分成左右两个部分。最后,我们使用`hstack()`函数将左部分、插入的列和右部分重新拼接成一个新的矩阵。
阅读全文