首页 > 解决方案 > 动态插入文件名和路径以在熊猫中保存 csv

问题描述

我需要动态保存我的pandas数据框。我已经成功地使用以下代码输出了一个具有静态名称的 CSV 文件:

export_csv = df.to_csv(r'static_name.csv', header=False , index=False)

但我未能动态地完成这项工作。使用下面的代码,我希望得到一个带有传入名称args.save_file_name.csv后缀的文件。但是,我没有得到任何结果。

import os
import argparse
import pandas as pd

parser = argparse.ArgumentParser()
parser.add_argument('file', help="this is the file you want to open")
parser.add_argument('save_file_name', help="the name of the file you want for the output CSV")

args = parser.parse_args()
print("file name:", args.file) # checking that this worked

...

# export csv
path = os.getcwd()
export_path = path + args.save_file_name + '.csv'
export_csv = df.to_csv(path_or_buf=export_path, header=False, index=False)

标签: python-3.xpandasarguments

解决方案


我认为,这个问题出在你的export_path变量中,这是不对的。以下代码应该可以完成这项工作。

export_path = os.path.join(path, args.save_file_name + '.csv')

推荐阅读