cython

1
import cython
2

3

4
@cython.locals(counts=cython.int[10], digit=cython.int)
5
def count_digits(digits):
6
    """
7
    >>> digits = '01112222333334445667788899'
8
    >>> count_digits(map(int, digits))
9
    [1, 3, 4, 5, 3, 1, 2, 2, 3, 2]
10
    """
11
    counts = [0] * 10
12
    for digit in digits:
13
        assert 0 <= digit <= 9
14
        counts[digit] += 1
15
    return counts
16