创建一个由1-20之间的数值组成的4行5列的数组,截取:(1)前面2行与所有列组成的数组(2)后面两列与全部行组成的数组(3)中间两行与左右两端的列组成的数组
时间: 2024-02-13 17:06:20 浏览: 42
js代码-将数组截取成不同长度的数组
好的,以下是代码实现:
```python
import numpy as np
# 创建4行5列的数组
arr = np.array([[1, 2, 3, 4, 5],
[6, 7, 8, 9, 10],
[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20]])
# 截取前面2行与所有列组成的数组
sub_arr1 = arr[:2, :]
# 截取后面两列与全部行组成的数组
sub_arr2 = arr[:, -2:]
# 截取中间两行与左右两端的列组成的数组
sub_arr3 = arr[1:3, [0, -1]]
print("原始数组:\n", arr)
print("截取前面2行与所有列组成的数组:\n", sub_arr1)
print("截取后面两列与全部行组成的数组:\n", sub_arr2)
print("截取中间两行与左右两端的列组成的数组:\n", sub_arr3)
```
输出结果:
```
原始数组:
[[ 1 2 3 4 5]
[ 6 7 8 9 10]
[11 12 13 14 15]
[16 17 18 19 20]]
截取前面2行与所有列组成的数组:
[[1 2 3 4 5]
[6 7 8 9 10]]
截取后面两列与全部行组成的数组:
[[ 4 5]
[ 9 10]
[14 15]
[19 20]]
截取中间两行与左右两端的列组成的数组:
[[ 6 10]
[11 15]]
```
阅读全文