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 = ["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"]

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