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

完善函数fun2，参数data是一个整型列表，把data中小于5的数据替换为1,剩余数据替换为20。 必须使用Numpy，返回结果为Numpy的ndarray类型。

def fun2(data=[9,2,6,4,2,6,7,8,2,10] ):
    """
    Arg:
        data : a list as input; e.g. [9,2,6,4,2,6,7,8,2,10]
    return a Numpy ndarray; e.g. [20  1 20  1  1 20 20 20  1 20]
    """
    return np.where((np.array(data))<5,1,20)