Entropy for a Split

import math

def entropy(counts):
    total = sum(counts)
    e = 0.0
    for c in counts:
        p = c / total
        if p > 0:
            e += p * math.log2(p)
    return e