Question

我还是初学者，我需要一点帮助。有一个代码可以为我计算风险价值。在main函数中有三个任务我需要输入filename：

df = pd.read_excel("data/WTS.xls")
AllVaR.to_excel("portfolio/WTS.xls")
plt.savefig('charts/WTS.png')

现在我必须为我的学术研究做250件。我将所有数据文件放在文件夹＆＃34; data＆＃34;我想知道如何为每个文件创建一个循环，以及如何将这些文件名放入pandas read，write和matplotlib函数中。

import os
for filename in os.listdir('data/'):
    df = pd.read_excel("data/filename?.xls")
    AllVaR.to_excel("portfolio/filename?.xls")
    plt.savefig('charts/filename?.png'

提前感谢您的帮助！

Answer 1

您可以使用glob和os：

import glob
import os
for f in glob.glob("data/filename*.xls"): 
     base = os.path.basename(f)
     basewoe = os.path.splitext(base)[0]

     df = pd.read_excel(f)
     AllVaR.to_excel("portfolio/"+ base)
     plt.savefig("charts/"+ basewoe +".png")

使用pandas和matplotlib循环遍历所有文件

1 个答案: