-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathd.py
85 lines (64 loc) · 2.35 KB
/
d.py
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
# from datetime import datetime, timedelta
# from collections import Counter
# import json as JSON
# raw_data = JSON.load(open("passengerRoutes.json"))
# print(type(raw_data))
# data = {
# 'pickup_location': [],
# 'drop_location': [],
# 'pickup_time': [],
# 'pick_drop': [],
# }
# for record in raw_data:
# time = record["time"]
# data['pickup_location'].append(record["routes"][0])
# data["drop_location"].append(record["routes"][1])
# data["pickup_time"].append(time)
# data['pick_drop'].append(f'{record["routes"][0]}-{record["routes"][1]}')
# grouped_data = {}
# dates = []
# for i in data['pickup_time']:
# dates.append(i.split(" ")[0])
# dates = sorted(list(set(dates)), key= lambda x : int(x.split("-")[2]))
# print(dates)
# for i in dates:
# grouped_data[i] = {
# 'pickup_location': [],
# 'drop_location': [],
# 'pickup_time': [],
# 'pick_drop': [],
# }
# for record in raw_data:
# time = record["time"]
# date = time.split(" ")[0]
# grouped_data[date]['pickup_location'].append(record["routes"][0])
# grouped_data[date]["drop_location"].append(record["routes"][1])
# grouped_data[date]["pickup_time"].append(time)
# grouped_data[date]['pick_drop'].append(f'{record["routes"][0]}-{record["routes"][1]}')
# data['pickup_time'] = [datetime.strptime(time, '%Y-%m-%d %H:%M:%S') for time in data['pickup_time']]
# data['pickup_day'] = [time + timedelta(days=1) for time in data['pickup_time']]
# next_day_data = {
# 'pickup_location': [],
# 'drop_location': [],
# 'pickup_time': [],
# 'pick_drop':[]
# }
# for i in range(len(data['pickup_time'])):
# next_day_data['pickup_location'].append(data['pickup_location'][i])
# next_day_data['drop_location'].append(data['drop_location'][i])
# next_day_data['pickup_time'].append(data['pickup_time'][i])
# next_day_data['pick_odrp'].append(data['pick_drop'][i])
# pickup_demand = Counter(next_day_data['pick_drop'])
# for i in pickup_demand.keys():
# print(i, pickup_demand[i])
# print(dict(pickup_demand))
from collections import Counter
import json as JSON
raw_data = JSON.load(open("passengerRoutes.json"))
data = {
'pick_drop': [],
}
for record in raw_data:
data['pick_drop'].append(f'{record["routes"][0]}-{record["routes"][1]}')
pickup_demand = Counter(data['pick_drop'])
print(dict(pickup_demand))