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

完善函数fun3，参数data是一个列表(可能有多维)，num是一个数字，返回data中与num差值最大的数。 (提示: np.abs, argmax, flat)

这里要注意以下，data可能有多维，所以可以先把data转换成一维，然后找到差值最大即可

def fun3(data=[1,2,3,4,5,6,7],num=5):
    """
    Arg:
        data : a list as input; e.g. [1,2,3,4,5,6,7]
    return a number; e.g. 1

    """
    data1=np.array(data)
    data2=np.array(list(data1.flat))
    diff=np.abs(data2-num)
    index=np.argmax(diff)
    return data2.flat[index]