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()