numpy二维数组中取出行列均为0,2,4的数据组成新的矩阵
时间: 2024-04-16 11:24:23 浏览: 135
python NumPy ndarray二维数组 按照行列求平均实例
可以使用 numpy 的索引功能来取出二维数组中行列均为 0、2、4 的数据,并组成新的矩阵。下面是一个示例:
```python
import numpy as np
# 创建一个二维数组
matrix = np.array([[1, 2, 3, 4, 5],
[6, 7, 8, 9, 10],
[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20],
[21, 22, 23, 24, 25]])
# 取出行列均为 0、2、4 的数据
new_matrix = matrix[[0, 2, 4]][:, [0, 2, 4]]
print("新的矩阵:")
print(new_matrix)
```
输出结果:
```
新的矩阵:
[[ 1 3 5]
[11 13 15]
[21 23 25]]
```
以上代码示例中,我们首先创建了一个二维数组 `matrix`。然后,使用索引 `[[0, 2, 4]][:, [0, 2, 4]]` 来取出行列均为 0、2、4 的数据,并赋值给新的矩阵 `new_matrix`。最后,打印出新的矩阵。这里的索引 `[0, 2, 4]` 表示选取行的索引,`[:, [0, 2, 4]]` 表示选取列的索引。
阅读全文