【python】一个纯数字字符串S,请返回连续单一数字字串的个数

题目：

"""
有一个纯数字组成的字符串, 返回连续单一数字子串的个数
输入字符串： “22252”
只含单一数字的子串是
1个字符：2出现4次，5出现1次  5
2个字符 22 出现2 次 3
3个字符 222 出现1 次
4个子串 0次
5个字符 0次
总共 4+1+2+1 =8
输出结果：8

示例:
输入：22252
输出： 8
"""
代码如下：

a = "22252"
def is_num(s):
    res = 0
    for i in range(len(s)-1):   # 遍历单个数字 2 2 2 5
        res += 1  
        x = s[i]
　　　　 for j in range(i + 1, len(s)):  # 遍历多个相同的数字
            if s[j] == x:       # 22 222 ; 22 ;
                res += 1　　
            else:
                break
    res += 1    # 当 i == 4时与len(i) - 1相等，程序循环不执行,少一个 res+=1
    return res
b = is_num(a)
print(b)　　# 8