PTA-Python-函数-3 使用函数统计指定数字的个数 (20分)

点击链接PTA-Python-AC全解汇总

题目：
本题要求实现一个统计整数中指定数字的个数的简单函数。

CountDigit(number,digit )

其中number是整数，digit为[1, 9]区间内的整数。函数CountDigit应返回number中digit出现的次数。

函数接口定义：
在这里描述函数接口。例如：

CountDigit(number,digit ),返回digit出现的次数

裁判测试程序样例：

/* 请在这里填写答案 */

number,digit=input().split()
number=int(number)
digit=int(digit)
count=CountDigit(number,digit )
print("Number of digit 2 in "+str(number)+":",count)

输入样例：

-21252 2

输出样例：

Number of digit 2 in -21252: 3

我的代码：

def CountDigit(number,digit):
    cnt=0
    for x in str(number):
        if x!='-' and int(x)==digit:
            cnt+=1
    return cnt