地铁刷卡数据分析第二关

#在上一关的基础上，利用每天最后一条刷卡记录的行标号，

#统计获得159站点每天的进站人数和出站人数，结果用一个数据框df来表示，字段名称依次为：

#A1_日期、A2_进站人数、A3_出站人数

def return_values():

    import pandas as pd

    import step9_1

    r=step9_1.return_values()

    A=pd.read_csv('159站点8月份刷卡数据.csv',sep=',',engine='python')

    A1=[]

    A2=[]

    A3=[]

    for i in range(len(r)):

        A1.append(r[i][0][:10])

        if i==0:

            At=A.iloc[:r[i][1]+1,:]

            A2.append(len(At.iloc[At.iloc[:,0].values==21,:]))

            A3.append(len(At.iloc[At.iloc[:,0].values==22,:]))

        if i>0:

            At=A.iloc[r[i-1][1]+1:r[i][1]+1,:]

            A2.append(len(At.iloc[At.iloc[:,0].values==21,:]))

            A3.append(len(At.iloc[At.iloc[:,0].values==22,:]))

    A1.append('2015-08-31') 

    At=A.iloc[r[len(r)-1][1]+1:,:] 

    A2.append(len(At.iloc[At.iloc[:,0].values==21,:]))

    A3.append(len(At.iloc[At.iloc[:,0].values==22,:]))

    D={'A1_日期':A1,'A2_进站人数':A2,'A3_出站人数':A3}

    df=pd.DataFrame(D)

    return df