From fe8b2ba0e7fc96a052bde94b871a5c9b1c30f99f Mon Sep 17 00:00:00 2001 From: Hannes Kuchelmeister Date: Tue, 4 May 2021 11:47:46 +0200 Subject: [PATCH] add party affiliation to uk data --- VotingAnalysis.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/VotingAnalysis.py b/VotingAnalysis.py index faf063a..e6ef1f2 100644 --- a/VotingAnalysis.py +++ b/VotingAnalysis.py @@ -55,6 +55,7 @@ def load_uk_data(): data = pd.DataFrame() name_column = 'Member' + party_column = 'Party' vote_column = 'Vote' column_to_filename = {} @@ -83,7 +84,7 @@ def load_uk_data(): if data.empty: # if first file that is loaded set data equal to data from first file - data = df[[name_column, vote_column_name]] + data = df[[name_column, party_column, vote_column_name]] else: # merge data with already loaded data data = data.merge(df[[name_column, vote_column_name]], on=name_column) @@ -262,7 +263,7 @@ plt.style.use('ggplot') # Load data data = load_uk_data().to_numpy() -X = data[:,1:] +X = data[:,2:] print(X) inp = X.shape[1] # No of features (bill count)