1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
# Takes valid-full.csv which is a subset of the lines of train.csv, formatted in the
# exact same way
# Outputs valid.csv which contains the polylines cut at an arbitrary location, and three
# new columns containing the destination point and the length in seconds of the original polyline
# (see contest definition for the time taken by a taxi along a polyline)
import random
import csv
import ast
with open("valid-full.csv") as f:
vlines = [l for l in csv.reader(f)]
def make_valid_item(l):
polyline = ast.literal_eval(l[-1])
last = polyline[-1]
cut_idx = random.randrange(len(polyline)+1)
cut = polyline[:cut_idx]
return l[:-1] + [
cut.__str__(),
last[0],
last[1],
15 * (len(polyline)-1),
]
vlines = map(make_valid_item, filter(lambda l: (len(ast.literal_eval(l[-1])) > 0), vlines))
with open("valid.csv", "w") as f:
wr = csv.writer(f)
for r in vlines:
wr.writerow(r)
with open("valid-solution.csv", "w") as f:
wr = csv.writer(f)
wr.writerow(["TRIP_ID", "LATITUDE", "LONGITUDE"])
for r in vlines:
wr.writerow([r[0], r[-2], r[-3]])
|
