python 在不同文件夾中查找相同文件名，並利用兩個文件完成某項任務

Question

不同文件夾中有文件夾如下

folder 1 - a, b, c...
folder 2 - a, b, c ...

每個文件夾都有類似名稱的文件，如下所示

folder 1 - a - one.csv, two.csv,...
folder 2 - a - one_pred.csv, two_pred.csv,...

我想檢查每個文件夾中的文件，當它找到相同的文件夾時，go 進入文件夾，當它找到one.csv & one_pred.csv時，例如，

它轉到 ML 性能檢查 function

rms = get_rms(one, one_pred)

但我不確定如何同時遍歷兩個不同的文件夾並將它們傳遞給 function。

我試過了

folders = os.listdir(folder1)
for d in folders:
    files = os.listdir(folder1 + d)
    for f in files:
        if f.split('_')[-1] == 'pred':
            f_truth = f.replace('_pred','')
            rms = get_rms(f_truth, f)

它沒有用。 在我運行它之前我認為這是錯誤的，但我不知道我應該做些什么不同......

Answer 1

使用 python 的os.walk()獲取兩個文件夾中的文件列表。 然后根據您的需要匹配它們並運行您想要的任何 function。 使用這個和這個參考來了解如何使用os.walk() 。