KL Divergence Direction

import math

def kl_divergence(p, q):
    # KL(p || q)
    return sum(pi * math.log(qi / pi) for pi, qi in zip(p, q))

print(kl_divergence([0.9, 0.1], [0.5, 0.5]))