python 基础升级 第八课 作业 瞌睡龙龙

from matplotlib import pyplot as plt
import numpy as np
import pandas as pd
from numpy import nan as NA
from pandas import DataFrame, Series
data_train = pd.read_csv('E:/pythondoc/2016-us-ge-by-county.csv')
s0=data_train.StateName.unique()        #列出51个选区
df_state_all=DataFrame()                #初始化总表
for i in range(len(s0)):

选出单个选区的所有投票情况

    df_state=data_train.loc[data_train.StateName==s0[i],['StateName','CountyTotalVote','Candidate','VoteCount']]   

这个选区的Trump的得票数

    s_state_T=df_state.VoteCount[df_state.Candidate=='Trump']  

这个选区的Clinton的得票数

    s_state_C=df_state.VoteCount[df_state.Candidate=='Clinton'] 

这个选区的总投票数

    s_state_tatol=df_state.CountyTotalVote[df_state.Candidate=='Trump']           

这个选区的相应票数小计,合并为表

    df_state1=DataFrame({'stateTotalVote':s_state_tatol.sum(),'state_T':s_state_T.sum(),'state_C':s_state_C.sum()},
                        index=[s0[i]],
                        columns=['stateTotalVote','state_T','state_C'])      

选区小计表合并到总表

    df_state_all=pd.concat([df_state_all,df_state1])                                                               
    
df_state_all
已邀请:

要回复问题请先登录注册

返回顶部