Implement file parsing

This commit is contained in:
coolneng 2021-03-15 18:19:59 +01:00
parent 09caafc2cf
commit 46efa13c5b
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 24 additions and 0 deletions

24
src/preprocessing.py Normal file
View File

@ -0,0 +1,24 @@
from sys import argv
from pandas import read_table
def read_header(filename):
with open(filename, "r") as f:
header = f.readline().split()
return header[0], header[1]
def parse_file(filename):
n, m = read_header(filename)
df = read_table(
filename, names=["source", "destination", "distance"], sep=" ", skiprows=[0]
)
return n, m, df
def main():
parse_file(argv[1])
if __name__ == "__main__":
main()