hsaest's picture
Upload folder using huggingface_hub
9be4956 verified
raw
history blame
856 Bytes
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]})