Implement file parsing
This commit is contained in:
		
							parent
							
								
									09caafc2cf
								
							
						
					
					
						commit
						46efa13c5b
					
				
							
								
								
									
										24
									
								
								src/preprocessing.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										24
									
								
								src/preprocessing.py
									
									
									
									
									
										Normal 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()
 | 
				
			||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user