Pythonで選挙データを分析してみよう！⑦〜サポーターと選挙活動の効果測定編〜

def main():
    # ... データ読み込み ...
    supporter_data = load_supporter_data()
    campaign_data = load_campaign_data()
    voting_results = load_voting_results()
    voting_info = load_voting_info()

    # 1. 投票結果データをベースにする
    merged_data = voting_results.copy()
    
    # 2. サポーターデータをマージ
    if supporter_data is not None:
        merged_data = merged_data.merge(
            supporter_data, 
            left_on='サポーター照合用', # 正規化した市区町村名
            right_on='市区町村', 
            how='left'
        )
        # サポーターがいない地域は 0 で埋める
        merged_data['サポーター数'] = merged_data['サポーター数'].fillna(0)

    # 3. 演説・街宣データをマージ
    if campaign_data is not None:
        merged_data = merged_data.merge(
            campaign_data, 
            left_on='街宣照合用', # 正規化した市区町村名
            right_on='市区町村', 
            how='left'
        )
        merged_data['演説街宣回数'] = merged_data['演説街宣回数'].fillna(0)

    # 4. 投票総数データをマージして、得票率を計算
    # ...

def calculate_correlations(merged_data):
    """
    相関関係を計算する関数
    """
    print("\n=== 相関分析結果 ===")
    
    # サポーター数と得票率の相関
    if 'サポーター数' in merged_data.columns and '得票率' in merged_data.columns:
        supporter_rate_corr = merged_data['サポーター数'].corr(merged_data['得票率'])
        print(f"サポーター数と得票率の相関係数: {supporter_rate_corr:.4f}")
    
    # 演説街宣回数と得票率の相関
    if '演説街宣回数' in merged_data.columns and '得票率' in merged_data.columns:
        campaign_rate_corr = merged_data['演説街宣回数'].corr(merged_data['得票率'])
        print(f"演説街宣回数と得票率の相関係数: {campaign_rate_corr:.4f}")

def analyze_campaign_effect(merged_data):
    """
    演説・街宣の効果を分析する関数
    """
    print("\n=== 演説・街宣効果分析 ===")
    
    # 演説・街宣があった地域と、なかった地域にデータを分割
    campaign_yes = merged_data[merged_data['演説街宣回数'] > 0]
    campaign_no = merged_data[merged_data['演説街宣回数'] == 0]
    
    if len(campaign_yes) > 0 and len(campaign_no) > 0:
        # それぞれ
        yes_avg_rate = campaign_yes['得票率'].mean()
        no_avg_rate = campaign_no['得票率'].mean()
        
        print(f"演説・街宣あり - 平均得票率: {yes_avg_rate:.2f}%")
        print(f"演説・街宣なし - 平均得票率: {no_avg_rate:.2f}%")
        print(f"得票率差: {yes_avg_rate - no_avg_rate:+.2f}%")