if CSV.split('\\')[-1] == (str(i)+'.csv'): print(CSV) df = pd.read_csv( CSV, index_col=0) #index_col用来指定表格的索引值 index_col=0 是将第一列作为索引 #print(df) set_off = np.array(df['开始城区'][0:len(df)].astype(str))#获取开始城区 arrive = np.array(df['结束城区'][0:len(df)].astype(str))
时间: 2024-04-18 17:31:44 浏览: 49
save-csv.py
您的代码片段中,首先使用 `CSV.split('\\')[-1]` 来获取文件路径中的文件名,然后将其与 `(str(i)+'.csv')` 进行比较。如果相等,则打印出文件名 `CSV`。
接下来,使用 `pd.read_csv()` 函数读取 `CSV` 文件,并将第一列作为索引值。
然后,使用 `np.array()` 函数将 '开始城区' 和 '结束城区' 列的值转换为字符串数组,并存储在 `set_off` 和 `arrive` 中。
阅读全文