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

add removal of speaker and independent

Browse Source
This commit is contained in:
Hannes Kuchelmeister
2021-05-23 21:36:07 +02:00
parent f3fecfdc12
commit b4ff4f3c43
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
voting_lib/load_data.py
View File

@@ -20,7 +20,9 @@ def load_german_data():
party_column_g = 'Fraktion/Gruppe' party_column_g = 'Fraktion/Gruppe'
name_column = 'Member' name_column = 'Member'
party_column = 'Party' party_column = 'Party'
parties_to_remove = ['Fraktionslos']
vote_column_to_title = {} vote_column_to_title = {}
voting_features = ['ja', 'nein', 'Enthaltung', 'ungültig'] voting_features = ['ja', 'nein', 'Enthaltung', 'ungültig']
@@ -45,6 +47,7 @@ def load_german_data():
df[vote_column_name] = df[voting_features].sum(axis=1) df[vote_column_name] = df[voting_features].sum(axis=1)
df=df.rename(columns={name_column_g:name_column,party_column_g:party_column}) df=df.rename(columns={name_column_g:name_column,party_column_g:party_column})
df=df[~df.Party.isin(parties_to_remove)]
period = df.iloc[0][period_column_g] period = df.iloc[0][period_column_g]
@@ -73,6 +76,8 @@ def load_uk_data(path):
name_column = 'Member' name_column = 'Member'
party_column = 'Party' party_column = 'Party'
vote_column = 'Vote' vote_column = 'Vote'
parties_to_remove = ['Speaker', 'Deputy Speaker', 'Independent']
column_to_filename = {} column_to_filename = {}
@@ -105,5 +110,6 @@ def load_uk_data(path):
# merge data with already loaded data # merge data with already loaded data
data = data.merge(df[[name_column, vote_column_name]], on=name_column) data = data.merge(df[[name_column, vote_column_name]], on=name_column)
data = data[~data.Party.isin(parties_to_remove)]
print(data) print(data)
return data return data