#! /usr/bin/python
# -*- coding:utf-8 -*-

import pandas as pd

def main():
    inpputfile = 'C:\\Users\\yt\\Desktop\\Program\\Date1030-1112.txt'
    try:
        #读取txt文件
        interaldata  = pd.read_table(inpputfile, header=None,names=['studentid','interval','countNum'],sep='\t')
        #排序
        #interaldata.sort_values(by='studentid', ascending=True)
        #interaldata.sort_index(axis=1, ascending=True)
        #数据转换和去重
        uniqdata = interaldata['studentid'].to_frame().drop_duplicates() 
        #分组，确定分组的组数
        grouped = interaldata.groupby(interaldata['interval'])
        mergerK = len(grouped['interval'].count())
        #联结-组数上+1<从0开始>
        #联结的重复列名处理 -- suffixes-追加到重叠列名末尾
        #middata = pd.merge( middata  , interaldata[ interaldata['interval']==2], on='studentid' ,how='left',suffixes=('_left','-right'))
        middata = pd.merge( uniqdata  , interaldata[ interaldata['interval']==1], on='studentid' ,how='left')
        for ange in range(2,(mergerK+1)):
            middata = pd.merge(middata,interaldata[ interaldata['interval']==ange], on='studentid' ,how='left')
        # 将NAN替换为0
        outdata =   middata.fillna(0)
        #列数
        KK = outdata.shape[1]  #outdata.shape[0]行数
        colodd= range(1,KK,2)
        #特定位置添加列表元素
        colodd.insert(0,0)
        colevent = range(0,KK,2)
        tmpdata = outdata.ix[:,colodd]
        #数据类型转换 
            resultdata = outdata.ix[:,colevent]
        resultdata.ix[:,range(1,resultdata.shape[1])] = resultdata.ix[:,range(1,resultdata.shape[1])] .astype('int')
        #保存数据-去掉列名
        resultdata.to_csv("C:\\Users\\yt\\Desktop\\Program\\submission.csv", index=False,header=False)
    except  IOError:
        print("错误文件")
    else:
        print('Processing')
    finally:
        try:
            print "again
        except  NameError:
        print "again"

if __name__ == "__main__":  
    main()
##本程序中异常处理功能报错，这部分先放这，待需要时在修正
01.#两个对象的列名不同，可以分别指定，例：pd.merge(df1,df2,left_on='lkey',right_on='rkey')
    #middata.rename(columns={'interval_x':'a', 'interval_y':'c'}, inplace = True)
02.
interaldata[0:2]            #选取多个行
interaldata.ix[[0,2,4,5,7],[1,2]] #ix 的功能就更强大了，它允许我们混合使用下标和名称进行选取
interaldata.irow(0)     #取data的第一行
interaldata. iloc(0)        #取data的第一列 
interaldata.iat[1,1]        #选取第二行第二列，用于已知行、列位置的选取
# iloc 使得我们可以对column使用slice（切片）的方法对数据进行选取