python第七章课后习题(10)

完善函数fun10,参数data是一个整型列表,把data中奇数的数据替换为1, 偶数替换为-1。
必须使用numpy,返回结果为Numpy的ndarray类型。

提示:np.where(condition, x, y)
注意where的用法就行

def fun10(data=[9,2,6,4,2,6,7,8,2,10]):
    """
    Arg:
        data : an array as input; e.g. array([9,2,6,4,2,6,7,8,2,10])
    return a Numpy ndarray; e.g. [ 1 -1 -1 -1 -1 -1  1 -1 -1 -1]
    """
    nup=np.array(data)
    return np.where(nup%2!=0,1,-1)
    pass

猜你喜欢

转载自blog.csdn.net/qq_53029299/article/details/115117798