请问怎么用Python函数写出来

s = input("请输入：")
def func(string):
    num_low = 0
    num_num = 0
    for i in string:
        if i.isdigit():
            num_num += 1
        elif i.islower():
            num_low += 1
        else:
            continue
    print("小写字母的个数：{}".format(num_low))
    print("数字的个数：{}".format(num_num))
func(s)