2017-09-14 52 views
-3

我正在写大函数,我需要使用输入文件名来输出输出文件名。我尝试了一些东西如何使用pandas.read_csv导入文件时获取文件名

import pandas as pd 
import os 
input_file = pd.read_csv('my_file.csv',header=None) 
input_file_name = os.basename(input_file) 

但我无法取回文件名。 如何在这里检索'my_file'?

回答

0
def do_job(input_file): 
    if not os.path.exists(input_file): 
     sys.stderr.write("Error: '%s' does not exist"%input_file) 
     sys.exit(1) 
    input = pd.read_csv(input_file,header=None) 

    # do many operations 

    # so file name is stored in handle 'input_file' 
    # I can give output file name using input_file 
    output_name = 'Results_' + input_file 
相关问题