Allenfenqu/bus_line_match.py

9 lines
418 B
Python

def pipei():
import pandas as pd
data_path = r''
df1 = pd.read_csv(data_path + 'bus_road_sequence.csv', header=None, encoding='gbk')
df2 = pd.read_csv(data_path + 'links_test1.csv')
replace_dict = df2.set_index(df2.columns[5])[df2.columns[11]].to_dict()
df1.iloc[:, 1:] = df1.iloc[:, 1:].applymap(replace_dict.get)
df1.to_csv(data_path + 'bus_line_partition.csv', header=None, index=False)