def plot_unique_trips(): from matplotlib import pyplot as plt trip_lookup = {} print("Loading map") road_map = Map("nyc_map4/nodes.csv", "nyc_map4/links.csv") print("Matching nodes") sizes = [] with open("sample.csv", "r") as f: reader = csv.reader(f) reader.next() for line in reader: trip = Trip(line) trip.num_occurrences = 1 trip.origin_node = road_map.get_nearest_node( trip.fromLat, trip.fromLon) trip.dest_node = road_map.get_nearest_node(trip.toLat, trip.toLon) if ((trip.origin_node, trip.dest_node) in trip_lookup): #Already seen this trip at least once trip_lookup[trip.origin_node, trip.dest_node].num_occurrences += 1 elif trip.origin_node != None and trip.dest_node != None: #Never seen this trip before trip_lookup[trip.origin_node, trip.dest_node] = trip sizes.append(len(trip_lookup)) plt.plot(range(len(sizes)), sizes) plt.xlabel("Inner Loop Iteration") plt.ylabel("L1 Error (sec)") fig = plt.gcf() fig.set_size_inches(20, 10) fig.savefig('test2png.png', dpi=100) #Make unique trips into a list and return new_trips = [trip_lookup[key] for key in trip_lookup] return new_trips
def plot_unique_trips(): from matplotlib import pyplot as plt trip_lookup = {} print("Loading map") road_map = Map("nyc_map4/nodes.csv", "nyc_map4/links.csv") print("Matching nodes") sizes = [] with open("sample.csv", "r") as f: reader = csv.reader(f) reader.next() for line in reader: trip = Trip(line) trip.num_occurrences = 1 trip.origin_node = road_map.get_nearest_node(trip.fromLat, trip.fromLon) trip.dest_node = road_map.get_nearest_node(trip.toLat, trip.toLon) if((trip.origin_node, trip.dest_node) in trip_lookup): #Already seen this trip at least once trip_lookup[trip.origin_node, trip.dest_node].num_occurrences += 1 elif trip.origin_node !=None and trip.dest_node != None: #Never seen this trip before trip_lookup[trip.origin_node, trip.dest_node] = trip sizes.append(len(trip_lookup)) plt.plot(range(len(sizes)), sizes) plt.xlabel("Inner Loop Iteration") plt.ylabel("L1 Error (sec)") fig = plt.gcf() fig.set_size_inches(20,10) fig.savefig('test2png.png',dpi=100) #Make unique trips into a list and return new_trips = [trip_lookup[key] for key in trip_lookup] return new_trips