python 自定义函数

弄一个叫masking的函数，收到输入的名字，将名字中间的字更改为*

输入姓名： 张一二
张*二

def masking():
    name=str(input("输入名字："))
    if len(name) ==2:
        print(name[0:1]+"*")
    elif len(name) >2:
        print(name[0:1]+(len(name)-2)*"*"+name[-2:-1])