forked from amakelov/parlamentaren-kontrol
/
pk_plots.py
232 lines (208 loc) · 10 KB
/
pk_plots.py
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rcParams, gridspec
do_plots = True
party_colors = {
'АБВ' : '#6c1d45',
'АТАКА' : '#00563f',
'ББЦ' : '#ffff00',
'БДЦ' : '#ffff00', # maybe change
'БДЦНС' : '#ffff00', # maybe change
'БНС' : '#ca7725',
'БСП' : '#d71921',
'БСПЗБ' : '#d71921', # maybe change
'БСПЛБ' : '#d71921', # maybe change
'ВОЛЯ' : '#00718f',
'ГЕРБ' : '#0056a7',
'ДПС' : '#0066b7',
'ДСБ' : '#02528a',
'КБ' : '#6c1d45', # maybe change
'НДСВ' : '#f7cb41',
'ОДС' : '#002d7b',
'ОП' : '#000000',
'ПФ' : '#088a85',
'РБ' : '#0070bb',
'РЗС' : '#f67700',
'СК' : '#02528a', # maybe change
'независим': '#dddddd'
}
##############################################################################
# Matplotlib configuration.
##############################################################################
rcParams['font.family'] = 'sans-serif'
rcParams['font.sans-serif'] = ['FreeSans']
rcParams['font.size'] = 8
rcParams['figure.figsize'] = (5, 3)
figsize_square = (5, 5)
figsize_square_small = (3, 3)
figsize_long = (10, 3)
rcParams['savefig.dpi'] = 90
rcParams['legend.fontsize'] = 'medium'
rcParams['text.antialiased'] = True
rcParams['patch.antialiased'] = True
rcParams['lines.antialiased'] = True
##############################################################################
# Per stenogram plots.
##############################################################################
def registration_figure(date, names, reg_presences, reg_expected):
"""Barchart: x:name vs y:stacked presences and absences."""
if not do_plots: return
datestr = date.strftime('%Y%m%d')
datestr_human = date.strftime('%d/%m/%Y')
absences = reg_expected - reg_presences
pos = np.arange(len(names))
width = 0.35
f = plt.figure()
f.suptitle(u'Регистрирани Депутати %s'%datestr_human)
gs = gridspec.GridSpec(3,5)
main = f.add_subplot(gs[:,0:-1])
p1 = main.bar(pos, reg_presences, width, color='g')
p2 = main.bar(pos, absences, width, color='r', bottom=reg_presences)
main.set_ylabel(u'Брой Депутати')
main.set_xticks(pos+width/2.)
main.set_xticklabels(names)
main.tick_params(axis='x', length=0)
main.set_xlim(-0.5*width, pos[-1]+1.5*width)
main.legend((p1[0], p2[0]), (u'Присъстващи', u'Отсъстващи'), loc='upper left', bbox_to_anchor=(1,1))
summ = f.add_subplot(gs[-1,-1])
pie_array = [np.sum(reg_presences), np.sum(absences)]
summ.pie(pie_array, colors=['g', 'r'])
summ.set_title(u'Общо')
f.savefig('generated_html/registration%s.png' % datestr)
plt.close()
def absences_figure(date, names, vote_absences, vote_absences_percent):
"""Time series chart: x:session nb vs y:absences, color:name"""
if not do_plots: return
datestr = date.strftime('%Y%m%d')
datestr_human = date.strftime('%d/%m/%Y')
vote_absences = np.column_stack(vote_absences)
vote_absences_percent = np.column_stack(vote_absences_percent)
f = plt.figure()
f.suptitle(u'Отсъствия по Време на Гласуване %s'%datestr_human)
gs = gridspec.GridSpec(2,5)
su = f.add_subplot(gs[0,:-1])
su.plot(vote_absences, alpha=0.8)
su.set_ylabel(u'Брой Депутати')
su.set_ylim(0)
su.set_xticks([])
su.legend(names, loc='upper left', bbox_to_anchor=(1,1))
sd = f.add_subplot(gs[1,:-1], sharex=su)
sd.plot(vote_absences_percent, alpha=0.8)
sd.set_ylabel(u'% от Партията')
sd.set_xlabel(u'хронологичен ред на гласуванията')
sd.set_ylim(0, 100)
sd.set_xticks([])
sd.set_yticks([25, 50, 75])
f.savefig('generated_html/absences%s.png' % datestr)
plt.close()
##############################################################################
# Per stenogram session plots.
##############################################################################
f = plt.figure() # XXX "ugly tweak"
gs = gridspec.GridSpec(3,5) # These were made global and reused,
main = f.add_subplot(gs[:,0:-1]) # giving 2x speedup. The `remove` calls
summ = f.add_subplot(gs[-1,-1]) # below are part of the same tweak.
title = f.suptitle('')
main.set_ylabel(u'Брой Депутати')
summ.set_title(u'Общо')
def session_votes_by_party_figure(date, i, party_names, yes, no, abstain, absences):
if not do_plots: return
datestr = date.strftime('%Y%m%d')
datestr_human = date.strftime('%d/%m/%Y')
pos = np.arange(len(party_names))
width = 0.35
global f, main, summ, title
title.set_text(u'Гласували Депутати %s гл.%d' % (datestr_human, i+1))
p1 = main.bar(pos, yes, width, color='g')
p2 = main.bar(pos, no, width, color='r', bottom=yes)
p3 = main.bar(pos, abstain, width, color='c', bottom=yes+no)
p4 = main.bar(pos, absences, width, color='k', bottom=yes+no+abstain)
main.set_xticks(pos+width/2.)
main.set_xticklabels(party_names)
main.tick_params(axis='x', length=0)
main.set_xlim(-0.5*width, pos[-1]+1.5*width)
main.legend((p1[0], p2[0], p3[0], p4[0]), (u'За', u'Против', u'Въздържали се', u'Отсъстващи'), loc='upper left', bbox_to_anchor=(1,1))
pie_array = [np.sum(yes), np.sum(no), np.sum(abstain), np.sum(absences)]
wedges, texts = summ.pie(pie_array, colors=['g', 'r', 'c', 'k'])
f.savefig('generated_html/session%svotes%s.png' % (datestr, i+1))
for o in [p1, p2, p3, p4] + wedges + texts:
o.remove()
##############################################################################
# Pie plots for MPs.
##############################################################################
def alltime_regs(present, absent, manual):
if not do_plots: return
f = plt.figure(figsize=figsize_square)
f.suptitle(u'Обобщение на Всички Регистрации.')
s = f.add_subplot(1,1,1)
pie_array = [present, absent, manual]
s.pie(pie_array, colors=['g', 'r', 'c'], labels=[u'Присъствия', u'Отсъствия', u'Ръчно\nзаписани'])
f.savefig('generated_html/alltimeregs.png')
plt.close()
def alltime_votes(y, n, abst, absent):
if not do_plots: return
f = plt.figure(figsize=figsize_square)
f.suptitle(u'Обобщение на Всички Гласове.')
s = f.add_subplot(1,1,1)
pie_array = [y, n, abst, absent]
s.pie(pie_array, colors=['g', 'r', 'c', 'k'], labels=[u'За', u'Против', u'Въздържали се', u'Отсъстващи'])
f.savefig('generated_html/alltimevotes.png')
plt.close()
def alltime_regs_singleMP((present, absent, manual), name, asciiname):
if not do_plots: return
f = plt.figure(figsize=figsize_square_small)
f.suptitle(u'Обобщение на Всички Регистрации на\n%s'%name)
s = f.add_subplot(1,1,1)
pie_array = [present, absent, manual]
s.pie(pie_array, colors=['g', 'r', 'c'], labels=[u'Присъствия', u'Отсъствия', u'Ръчно\nзаписани'])
f.savefig('generated_html/alltimeregs_%s.png'%asciiname)
plt.close()
def alltime_votes_singleMP_compare_all((w, a), (abst, absent), name, asciiname):
if not do_plots: return
f = plt.figure(figsize=figsize_square_small)
f.suptitle(u'Обобщение на Всички Гласове на\n%s'%name)
s = f.add_subplot(1,1,1)
pie_array = [w, a, abst, absent]
s.pie(pie_array, colors=['g', 'r', 'c', 'k'], labels=[u'Съгласие с\nмнозинството', u'\n\n Противоречие с\n мнозинството', u'Въздържал се\n', u'Отсъстващ'])
f.savefig('generated_html/alltimevotes_compare_all_%s.png'%asciiname)
plt.close()
def alltime_votes_singleMP_compare_party((w, a), (abst, absent), name, asciiname):
if not do_plots: return
f = plt.figure(figsize=figsize_square_small)
f.suptitle(u'Обобщение на Всички Гласове на\n%s'%name)
s = f.add_subplot(1,1,1)
pie_array = [w, a, abst, absent]
s.pie(pie_array, colors=['g', 'r', 'c', 'k'], labels=[u'Съгласие с\nпартията', u'\n\n Противоречие с\n партията', u'Въздържал се\n', u'Отсъстващ'])
f.savefig('generated_html/alltimevotes_compare_party_%s.png'%asciiname)
plt.close()
def evolution_of_votes_singleMP(dates, votes, wa_all, wa_party, name, asciiname):
if not do_plots: return
f = plt.figure(figsize=figsize_long)
f.suptitle(u'Гласове и отсъствия на %s през годините.'%name)
absences = f.add_subplot(3,1,3)
with_all = f.add_subplot(3,1,1, sharex=absences)
with_party = f.add_subplot(3,1,2, sharex=absences)
all_votes_no_abs = np.sum(votes[:,:3], 1)
all_votes = np.sum(votes, 1)
mask_no_abs = np.logical_not(all_votes_no_abs)
mask = np.logical_not(all_votes)
with_all_array = np.ma.masked_array(100*wa_all[:,0], mask=mask_no_abs)/all_votes_no_abs
with_party_array = np.ma.masked_array(100*wa_party[:,0], mask=mask_no_abs)/all_votes_no_abs
absences_array = np.ma.masked_array(100*votes[:,3], mask=mask)/all_votes
with_all.plot(dates, with_all_array, '.-', alpha=0.3, linewidth=0.1)
with_all.legend([u'% съгласие с мнозинството (без отсъствия)'])
with_party.plot(dates, with_party_array, '.-', alpha=0.3, linewidth=0.1)
with_party.legend([u'% съгласие с партията (без отсъствия)'])
absences.plot(dates, absences_array, '.-', alpha=0.3, linewidth=0.1)
absences.legend([u'% отсъствия'])
with_all.set_yticks([25, 50, 75])
with_party.set_yticks([25, 50, 75])
absences.set_yticks([25, 50, 75])
with_all.set_ylim(0, 100)
with_party.set_ylim(0, 100)
absences.set_ylim(0, 100)
absences.set_xlim(dates[0], dates[-1])
f.autofmt_xdate()
f.savefig('generated_html/vote_evol_%s.png'%asciiname)
plt.close()