完善函数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)