Separates Michigan voter data into democratic, republican, and independent votes

election_data.py

import os
import pandas as pd

folder = './elections/'

for file in os.listdir(folder):
    df = pd.read_csv(folder + file, sep='\t', skipfooter=1)

    df['OfficeDescription'].loc[1]

    df = df.loc[df['OfficeDescription'] == 'ELECTORS OF PRESIDENT AND VICE-PRESIDENT OF THE UNITED STATES 4 YEAR TERM (1) POSITION']
    
    def g(df):
        pivoted = df.pivot_table(values='CandidateVotes', index='CountyName', columns='PartyDescription', aggfunc='sum')#+
        pivoted.columns = [col.replace(' ', '_') for col in pivoted.columns]
        return pivoted

    result = g(df.copy())
    
    df = result

    columns_to_sum = df.columns[~df.columns.isin(['REPUBLICAN', 'DEMOCRATIC', 'CountyName'])]

    df['Independents'] = df[columns_to_sum].sum(axis=1)

    df.drop(['GREEN', 'LIBERTARIAN', 'NATURAL_LAW', 'NO_PARTY_AFFILIATION', 'U.S._TAXPAYERS'], axis=1, inplace=True)

    df.to_csv(file.replace('.txt', '') + '_CLEANED.csv', index=False)