From e302ec0956e1ea1814f0aad74dc8318f96055eba Mon Sep 17 00:00:00 2001 From: Hannes Kuchelmeister Date: Tue, 4 May 2021 12:14:24 +0200 Subject: [PATCH] add party plot colors match mp plot --- VotingAnalysis.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/VotingAnalysis.py b/VotingAnalysis.py index 7baaa35..60cb343 100644 --- a/VotingAnalysis.py +++ b/VotingAnalysis.py @@ -186,6 +186,7 @@ def plot_mps(names, xs, ys, party_affiliation): def plot_parties(xs, ys, party_affiliation): + cmap = plt.cm.RdYlGn # converting parties to numeric format party_index_mapping, party_ids = np.unique(party_affiliation, return_inverse=True) @@ -201,7 +202,8 @@ def plot_parties(xs, ys, party_affiliation): party_ys /= party_count plt.figure() - plt.scatter(party_xs, party_ys) + party_colors=np.array(range(len(party_index_mapping))) + plt.scatter(party_xs, party_ys, c=party_colors, cmap=cmap) # plotting labels offset = 0.01 for x,y, party in zip(party_xs, party_ys, party_index_mapping):