def fetch_pairs(file, output='final/riderhorse.csv'): rows = read_rows([file]) # 32:rider, 33:horse merged = [] for row in rows: merged.append({'rider':row[32],'horse':row[33]}) unique = unique_pairs(merged) writer = csv.writer(open(output, "wb")) writer.writerow( ['Rider ID', 'Horse ID'] ) writer.writerows(unique) print len(unique)
def main(offset): #print "lets rank this shit" horse_rider = read_rows(['final/riderhorse.csv']) #print len(horse_rider) index = offset total = len(horse_rider) for i in range(index, total): entry = {'rider': horse_rider[i][0],'horse': horse_rider[i][1], 'years': []} for year in YEARS: entry['years'].append({'year': year, 'periods': search(year, horse_rider[i][0], horse_rider[i][1])}) index += 1 save(entry, "ranking/%d_rider_horse_%s_%s.csv" % (index, horse_rider[i][0], horse_rider[i][1])) print index