-
Notifications
You must be signed in to change notification settings - Fork 0
/
nodes_generator.py
34 lines (26 loc) · 1.02 KB
/
nodes_generator.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
import random
import numpy as np
import csv
xs = []
ys = []
class NodeGenerator:
def __init__(self, width, height, nodesNumber):
self.width = width
self.height = height
self.nodesNumber = nodesNumber
def generate(self):
# xs = np.random.randint(self.width, size=self.nodesNumber)
# ys = np.random.randint(self.height, size=self.nodesNumber)
with open('Locations.csv') as csv_file:
csv_reader = csv.reader(csv_file, delimiter=',')
line_count = 0
for row in csv_reader:
if line_count == 0:
print(f'Column names are {", ".join(row)}')
line_count += 1
else:
# print(f'\t{row[0]} ***** {row[1]} ***** {row[2]}.')
xs.append(float(row[2]))
ys.append(float(row[1]))
line_count += 1
return np.column_stack((xs, ys))