import re | |
import pandas as pd | |
import csv | |
def extract_before_parenthesis(s): | |
match = re.search(r'^(.*?)\([^)]*\)', s) | |
return match.group(1) if match else s | |
if __name__ == '__main__': | |
data = pd.read_csv('/home/xj/toolAugEnv/code/toolConstraint/database/googleDistanceMatrix/distance.csv') | |
data = data.to_dict(orient = 'split') | |
fieldnames = ['origin', 'destination', 'cost', 'duration', 'distance'] | |
with open('/home/xj/toolAugEnv/code/toolConstraint/database/googleDistanceMatrix/distance2.csv', 'w', newline='') as csvfile: | |
writer = csv.DictWriter(csvfile, fieldnames=fieldnames) | |
writer.writeheader() | |
for row in data['data']: | |
writer.writerow({'origin': extract_before_parenthesis(row[0]), 'destination': extract_before_parenthesis(row[1]), 'cost': row[2], 'duration': row[3], 'distance': row[4]}) |