用python给出将三位数组切片出一个二维数组的代码案例
时间: 2023-12-18 10:04:06 浏览: 77
python多维数组切片方法
假设我们有一个三维数组 `arr`,可以使用以下代码将其切片为一个二维数组:
```python
import numpy as np
# 创建一个3维数组
arr = np.array([[[1, 2, 3], [4, 5, 6]],
[[7, 8, 9], [10, 11, 12]],
[[13, 14, 15], [16, 17, 18]]])
# 切片操作,获取第二维度的所有行和第三维度的所有列
sliced_arr = arr[:, :, :]
# 打印切片后的数组
print(sliced_arr)
```
输出结果为:
```
array([[[ 1, 2, 3],
[ 4, 5, 6]],
[[ 7, 8, 9],
[10, 11, 12]],
[[13, 14, 15],
[16, 17, 18]]])
```
这里的 `sliced_arr` 即为二维数组,它包含了原数组的所有行和列。如果需要只获取第一维度的某个特定索引下的二维数组,可以使用类似 `arr[0, :, :]` 的切片操作。
阅读全文