Numpy初体验--第3关：Numpy数组的切片与索引

任务描述
本关的小目标是，从数组中选择指定的元素。

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

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

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

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

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

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

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

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

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

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

# 引入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 *********#
    #********** End **********#
    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