將代碼運行到一個文件夾中的多個文件

Question

我寫了一個開頭的代碼

import pandas as pd
# Code starts by reading one txt file
df=pd.read_fwf(r'C:\Users\youngrakchoi\Desktop\.filename.txt',sep='')

#-------My code--------#

# Code ends with saving the results
VOLTAGE_fin.to_excel(r'C:\Users\youngrakchoi\Desktop\NAME.xlsx')

現在我想將此代碼運行到保存在特定目錄（文件夾）中的所有文件並將文件保存為原始文件名（例如，將 VOLTAGE_fin 保存為“文件名”）。

有人可以幫我解決這個問題嗎？

Answer 1

我認為它會完成這項工作。

my_var_name = [ k for k,v in locals().items() if v == VOLTAGE_fin][0]
VOLTAGE_fin.to_excel(f"C:\\Users\\youngrakchoi\\Desktop\\{my_var_name}.xlsx")

Answer 2

我正在嘗試將我的編碼應用於一個文件夾中的多個文件

您可以使用os模塊中的listdir function 讀取給定目錄中的文件（它們的名稱）。 一個最小的例子如下：

import os
path = 'C:\\Users\\youngrakchoi\\Desktop\\my_dir\\'
all_files = os.listdir(path)
for file_name in all_files:
    ## Do your work 
    ## Finally save your output in xlsx format as follows
    pd.to_excel("{}/{}.xlsx".format(path, file_name))