使用 ArcGIS Pro 脚本工具将 pandas DataFrame 导出到 csv 文件

Question

我在 ArcGIS Pro 中创建了一个脚本工具，要求用户选择一个 csv 文件，然后将该 csv 文件读入 pandas DataFrame 并执行一些计算。计算完成后，我希望能够在与输入 csv 文件相同的文件夹中导出新的 csv 文件。

例如，用户从（C:\Users\Luis\GIS\input.csv）中选择输入csv文件。 我想在脚本末尾导出一个新的 csv 到 (C:\Users\Luis\GIS\output.csv)

用户输入读取的参数设置如下：

input_csv = arcpy.GetParameterAsText(0)

这个参数的数据类型是File，是ArcGIS Pro中的Tool Properties

Answer 1

如果你只想获取输入文件所在的文件夹，并且 input_csv 参数是绝对路径，你可以使用以下

input_csv = arcpy.GetParameterAsText(0)
output_csv = input_csv.split('\\')[:-1]
output_csv = '\\'.join(output_csv + ['output.csv'])

然后使用新的文件路径保存您的 DataFrame

df.to_csv(output_csv)