Adapt preprocessing module to the new lab
This commit is contained in:
parent
103a5bdfb0
commit
c5bccf2f29
|
@ -4,11 +4,12 @@ from pandas import read_table
|
|||
def read_header(filename):
|
||||
with open(filename, "r") as f:
|
||||
header = f.readline().split()
|
||||
return int(header[0]), int(header[1])
|
||||
return int(header[0])
|
||||
|
||||
|
||||
def parse_file(filename):
|
||||
n, m = read_header(filename)
|
||||
n = read_header(filename)
|
||||
m = 50
|
||||
df = read_table(
|
||||
filename, names=["source", "destination", "distance"], sep=" ", skiprows=[0]
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue