nodeExtractor/plottingZurichMap.py at master · timolex/nodeExtractor · GitHub

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
import matplotlib.pyplot as plt
from imageio import imread
import csv

x, y, gwX, gwY = [], [], [], []

with open('coordinatesEDs.csv') as csvFile:
	readCSV = csv.reader(csvFile, delimiter=',')
	# skipping the first row
	next(csvFile)

	for row in readCSV:
		x.append(float(row[0]) / 1000)
		y.append(float(row[1]) / 1000)

with open('coordinatesGWs.csv') as csvFile:
	readCSV = csv.reader(csvFile, delimiter=',')
	# skipping the first row
	next(csvFile)

	for row in readCSV:
		gwX.append(float(row[0]) / 1000)
		gwY.append(float(row[1]) / 1000)

img = imread('Zurich.png')

plt.figure(dpi=400)
plt.xlabel('km')
plt.ylabel('km')
plt.scatter(x, y, zorder=1, color='darkred', s=10)
plt.scatter(gwX, gwY, zorder=2, color='blue', s=20)
plt.imshow(img, zorder=0, extent=[0.0, 20.0, 0.0, 20.0])

plt.savefig("Zurich_TTN_nodes.png")
plt.show()