31 lines
1.2 KiB
Python
31 lines
1.2 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.split(separator)
|
||
|
|
node1 = int(split[0])
|
||
|
|
node2 = int(split[1])
|
||
|
|
if not is_directed:
|
||
|
|
if not has_weight:
|
||
|
|
edges[(node1, node2)] = 1
|
||
|
|
else:
|
||
|
|
edges[(node1, node2)] = 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")
|