2017-09-14 5 views
-3

J'écris une grande fonction, j'ai besoin d'utiliser le nom du fichier d'entrée dans le but de donner le nom du fichier de sortie. J'ai essayé quelque chosecomment obtenir le nom de fichier lorsque le fichier est importé en utilisant 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) 

mais je ne peux pas récupérer le nom du fichier. Comment puis-je récupérer 'my_file' ici?

Répondre

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