mds-riksdagen/prepare_data.py at master · ragulin/mds-riksdagen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import csv
import sys
vote_ids = []
# voters
voters = {}
def encode_vote(vote):
  if vote == "Ja":
    return  1
  elif vote == "Nej":
    return -1
  else:
    return 0

with open(sys.argv[1], 'rb') as csv_file:
  reader = csv.reader(csv_file, delimiter=',')
  for row in reader:
    vote_id = row[2]
    name = row[4]
    voter_id = row[5]
    party = row[6]
    vote = row[8]

    if vote_id not in vote_ids:
      vote_ids.append(vote_id)

    if voter_id not in voters:
      vote_info = {
          'party': party,
          'name': name,
          'votes': {}
          }
      voters[voter_id] = vote_info

    if vote_id not in voters[voter_id]:
      voters[voter_id]['votes'][vote_id] = encode_vote(vote)


print "voter,name,party," + ",".join(vote_ids)
for voter_id, voter_info in voters.iteritems():
  row = []
  row.append(voter_id)
  row.append("\"" + voter_info['name'] + "\"")
  row.append(voter_info['party'])
  for vote_id in vote_ids:
    if vote_id in voter_info['votes']:
      row.append(str(voter_info['votes'][vote_id]))
    else:
      row.append("0")

  print ",".join(row)