import seaborn as sb
import numpy as np

ballots = np.loadtxt("./ballots1.txt", dtype=int, delimiter=" ", skiprows=1, max_rows=443, usecols=range(1,25))

data = np.zeros((24, 24), dtype=int)

for ballot in ballots:
    for i in range(len(ballot)):
        data[i][ballot[i]-1] += 1

positions = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24]
candidates = np.array(["GaetanLepage","JulienMalka","K900","Scrumplex","aanderse",
              "adamcstephens","cafkafk","crertel","djacu","drupol","getchoo",
              "jopejoe1","leona-ya","mschwaig","niklaskorz","nim65s",
              "nyabinary","philiptaron","pinpox","pluiedev","rhendric",
              "samueldr","tomberek","tomodachi94"])

avg = np.zeros(24)
for i in range(24):
    avg += (i+1) * data[i]
avg /= 443

sortkey = np.argsort(avg, axis=0)

for i in range(24):
    data[i] = data[i][sortkey]
candidates = candidates[sortkey]

sb.heatmap(data, xticklabels=candidates, yticklabels=positions)