diary20180625

import csv,geohash  

l = list()

file2 = open("target.csv","w",newline='')
writer = csv.writer(file2)


file1 = open("train.csv","rt")
reader=csv.reader(file1)
next(reader)
writer.writerow(['orderid','userid','bikeid','biketype','startdate','starttime','geohashed_start_loc_long','geohashed_start_loc_lat','geohashed_end_loc_long','geohashed_end_loc_lat'])
for line in reader:
    for index,data in enumerate(line):
        if(index<=3):
            l.append(data)
        if(index==4):
            l.append(data.split()[0])
            l.append(data.split()[1])
        if(index==5):
            l.append(geohash.decode(data)[0])
            l.append(geohash.decode(data)[1])
        if(index==6):
            l.append(geohash.decode(data)[0])
            l.append(geohash.decode(data)[1])
    writer.writerow(l)
    l=[]

file1.close()
file2.close()