add party data to dataset loading

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)