Entropy of a Distribution

import math

def entropy(p):
    # Shannon entropy in nats
    return sum(pi * math.log(pi) for pi in p)

print(entropy([0.5, 0.5]))  # expected ~0.693