index out of range

Написал код для подсчёта энтропии. Во время удаления элемента равного нулю пишет out of range. Я не могу понять, почему ему не хватает.

from math import log

def entropy():
    s = input()
    a = '0123456789'
    am = []
    pa = []
    for i in range(len(a)):
        am.append(s.count(a[i]))
        if am[i] == 0:
            am.remove(am[i])
    for i in range(len(am)):
        p = float(am[i]/len(s))
        l = log(p, len(a))
        f = l * p
        pa.append(f)
    w = sum(pa)
    if w < 0:
        w = w * -1
    print(w)

entropy()

Ответы (0 шт):