33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
import random as rand
|
|
|
|
|
|
def read_graph_from_file(path, separator=",", read_first_line=False, is_directed=False, has_weight=False):
|
|
edges = dict()
|
|
with open(path, 'r') as file:
|
|
read_line = read_first_line
|
|
line = file.readline()
|
|
while line != '':
|
|
if read_line:
|
|
if line[0] != "#":
|
|
split = line.strip('\n').split(separator)
|
|
node1 = int(split[0])
|
|
node2 = int(split[1])
|
|
if not is_directed:
|
|
if node1 < node2:
|
|
node1, node2 = node2, node1
|
|
if not has_weight:
|
|
edges[(node1, node2)] = 1
|
|
else:
|
|
edges[(node1, node2)] = int(split[2])
|
|
else:
|
|
read_line = True
|
|
line = file.readline()
|
|
return edges
|
|
|
|
|
|
def add_weight_to_file(graph, path, extension, separator=",", weight_scale=(1, 100)):
|
|
with open(path + "_with_weight" + extension, 'w') as file:
|
|
for key in graph.keys():
|
|
file.write(str(key[0]) + separator + str(key[1]) + separator + str(rand.randint(weight_scale[0],
|
|
weight_scale[1])) + "\n")
|