Calculate entropy

calculate_entropy.py

#!/usr/bin/python

import sys
import math

f = sys.argv[1]

_file = open(f)

data = {}
total = 0

for l in _file:
    d = l.split()
    data[d[0]] = float(d[1])
    total = float(d[1]) + total

E = 0;

for country in data:
    count = data[country]
    p = count/total
    #print "p {0} count {1}".format(p, count)
    E = E + p * (-1)* math.log(p,2)


bound = math.log(len(data),2)
print "Entrophy: {0} Upper bound {1} : ".format(E,bound)