python第七章课后习题plus(1)

完善函数fun1,参数data是一个整型列表,将data中为偶数的数字抽取出来,并存入到一个数组中。 必须使用Numpy,返回结果为Numpy的ndarray类型。

找到偶数索引就行

def fun1(data=[1,2,3,4,5,6,7,8,9,10]):
    """
    Arg:
        data : a list as input; e.g. [1,2,3,4,5,6,7,8,9,10]
    return a Numpy ndarray; e.g. [ 2  4  6  8 10]
    """
    data1=np.array(data)
    index=np.where(data1%2==0)
    return data1[index]

猜你喜欢

转载自blog.csdn.net/qq_53029299/article/details/115279336
今日推荐