第3关：Numpy 数组的切片与索引

任务描述

本关任务：从数组中选择指定的元素。

相关知识

为了完成本关任务，你需要掌握：数组的切片与索引。

数组的切片与索引

一维 Numpy 数组的切片操作与 Python 列表的切片一样。下面首先来定义数字 0 1 2 直到 8 的数组，然后通过指定下标 3 到 7 来选择数组的部分元素，这实际上就是提取数组中值为 3 到 6 的元素。

1. In: import numpy as np
2. In: a = np.arange(9)
3. In: a[3:7]
4. Out: array([3,4,5,6])

同时用下标选择元素，下标范围从 0 到 7，并且下标每次递增 2，如下所示：

1. In: a[:7:2]
2. Out:array([0,2,4,6])

也可以像 Python 数组一样，用负值下标来反转数组：

1. In: a[::-1]
2. Out: array([8,7,6,5,4,3,2,1,0])

对于二维数组的索引，类似与 Python 数组的列表：

1. In: a=np.array([[1,2,3],[4,3,2]])
2. In: a[1][0]
3. Out:array([4])
4. In:a[1,:2] #获取第1维的前2个元素
5. Out:array([4, 3])

编程要求

本关的任务是，补全右侧编辑器 Begin-End 内的代码，以实现 Numpy 数组的索引功能的功能。具体要求如下：

• 函数接受两个参数，然后返回切片找出的指定元素；
• 本关的测试样例参见下文；
• 本关设计的代码文件 ce.py 的代码框架如下：

1. # 引入numpy库
2. import numpy as np
3. # 定义cnmda函数
4. def ce(a,m,n):
5. '''
6. 参数：
7. a:是一个Numpy数组
8. m:是第m维数组的索引
9. n:第m维数组的前n个元素的索引
10. 返回值:
11. ret: 一个numpy数组
12. '''
13. ret = 0
14. # 请在此添加切片的代码,实现找出多维数组a中第m个数组的前n个元素并赋值给ret
15. #********** Begin *********#
16. #********** End **********#
17. return ret

测试说明

本关的测试过程如下:

• 平台运行 step2/cetest.py 文件，并以标准输入方式提供测试输入；

• cetest.py 文件调用 ce 中的 ce 方法，平台获取 cetest.py 的输出，然后将其与预期输出作对比，如果一致，则测试通过；否则测试失败。

• 预处理的数组

[[1,2,3,4,5],[7,8,9,10,11],[12,13,14,15]]

以下是平台对 step2/cetest.py 的测试样例：

测试输入：
1； 3； 预期输出：

[7,8,9]

测试输入： 2； 2； 预期输出：

[12,13]

代码：

# 引入numpy库
import numpy as np
# 定义cnmda函数
def ce(a,m,n):
    '''
    参数：
    a:是一个Numpy数组
    m:是第m维数组的索引
    n:第m维数组的前n个元素的索引
    返回值:
    ret: 一个numpy数组
    '''
    ret = 0
    # 请在此添加切片的代码,实现找出多维数组a中第m个数组的前n个元素 并赋值给ret
    #********** Begin *********#
    ret = a[m,:n]
    #********** End **********#
    return ret