13hannes11/UU_NCML_Project
1
0
Fork 0
mirror of https://github.com/13hannes11/UU_NCML_Project.git synced 2024-09-03 20:50:59 +02:00
Code Issues Projects Releases Wiki Activity

plot parties based on mp avergae position #9

Browse Source
This commit is contained in:
Hannes Kuchelmeister
2021-05-04 11:10:22 +02:00
parent 0587711e68
commit bbb5bf13bf
1 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
VotingAnalysis.py
View File

@@ -219,9 +219,30 @@ ys, xs = np.unravel_index(np.argmax(X, axis=1), (h, w))
ys_disp = ys + np.random.rand(ys.shape[0])
xs_disp = xs + np.random.rand(xs.shape[0])
party_index_mapping, party_affiliation_numeric = np.unique(data[:,1], return_inverse=True)
party_index_mapping, party_ids = np.unique(data[:,1], return_inverse=True)
plot_mps(fig, ax, xs_disp, ys_disp, data[:,0] + " (" + data[:,1] + ")", party_ids)
plt.show()
party_count = np.zeros(party_index_mapping.shape[0])
party_xs = np.zeros(party_index_mapping.shape[0])
party_ys = np.zeros(party_index_mapping.shape[0])
for x, y, party_id in zip(xs, ys, party_ids):
party_xs[party_id] += x
party_ys[party_id] += y
party_count[party_id] += 1
party_xs /= party_count
party_ys /= party_count
plt.figure()
plt.scatter(party_xs, party_ys)
# plotting labels
offset = 0.01
for x,y, party in zip(party_xs, party_ys, party_index_mapping):
plt.text(x + offset, y + offset, party)
plot_mps(fig, ax, ys_disp, xs_disp, data[:,0] + " (" + data[:,1] + ")", party_affiliation_numeric)
plt.show()
#Simple SOFM for UK