1021. 个位数统计 (15)------Python

最近在使用python，所以找了pat的题练手，第一次在pat上运行python程序，因为一个小小的格式问题，浪费了我好长时间，真是够了。最后发现就是输出的格式与题目中国的不符，我输出按照习惯多加了两个空格，真的是醉了。总之，这个题目还是偏简单的，使用python也是很好的就能实现。

贴上代码，供大家参考吧，有啥意见的可以尽管提出了，大家一块进步。
def count_number(number, list):
count = 0
for num in list:
if num == str(number):
count = count + 1
return count

my_number = input()
list = list(my_number)
flag = 0

while True:
if flag == 10:
break
else:
count = count_number(flag, list)

if count != 0:
print(str(flag) + ":" + str(count))

flag += 1