From 7ca3762e43732a95c9500995b8e5ebaf77cadbf1 Mon Sep 17 00:00:00 2001 From: Hannes Kuchelmeister Date: Tue, 4 May 2021 10:41:42 +0200 Subject: [PATCH] add party data to dataset loading --- VotingAnalysis.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/VotingAnalysis.py b/VotingAnalysis.py index 732454f..3ae3765 100644 --- a/VotingAnalysis.py +++ b/VotingAnalysis.py @@ -16,6 +16,7 @@ def load_german_data(): data = pd.DataFrame() name_column = 'Bezeichnung' + party_column = 'Fraktion/Gruppe' vote_column_to_title = {} @@ -38,7 +39,7 @@ def load_german_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) @@ -176,7 +177,7 @@ plt.style.use('ggplot') # Load data data = load_german_data().to_numpy() -X = data[:,1:] +X = data[:,2:] print(X) inp = X.shape[1] # No of features (bill count)