Python计算机二级考试备考之统计字符串中不同类型字符的个数

#编写一个程序计算输入字符串中数字、字母、空格以及其他字符的个数
def countStr():
    a=input("请输入一个字符串：")
    N=M=X=Y=0
    for i in a:
        if str.isdigit(i)==True:
            N+=1
        elif str.isalpha(i)==True:
            M+=1
        elif str.isspace(i)==True:
            X+=1
        else:
            Y+=1
    return N,M,X,Y
A,B,C,D=countStr()
print("这个字符串中包含{}个数字，{}个字母，\
{}个空格，{}个其他字符".format(A,B,C,D))