使用python遍歷文件夾取出特定的字符串

# -*- coding: utf-8 -*
import re
import os
# 需要處理的文件夾路徑（絕對路徑）
path = u"/Users/a140/Downloads/test2"
# 存放結果的文件路徑（絕對路徑）
results = u"/Users/a140/Downloads/result.txt"
# 載入文件列表
file_list = os.listdir( path )
# 文件排序
file_list.sort()
# 遍歷所有文件
for file in file_list:
    # 打開要輸出的文件
    fo = open( path + '/' + file, "r")
    # 輸出文件名
    print("fileName: ", fo.name)
    # 打開要寫入結果的文件
    resu = open(results, "a+")
    # 寫入文件名和換行
    resu.write(fo.name + '\n')
    # 遍歷文件內的每一行
    for line in fo.readlines():
        # 判斷當前行是否包含"TOTAL:"
        if len(re.findall('TOTAL:', line)) != 0:
            # 若包含則將此行賦值給變量tmp
            tmp = line
    # 關閉文件
    fo.close()
    # 取出TOTAL值
    TotalLine = tmp.split('TOTAL SWAP')[0].split(': ')
    # 輸出TOTAL值
    print('Name: {}, TotalNum:{}'.format(TotalLine[0], TotalLine[1]))
    # 將TOTAL值寫入結果的文件
    resu.write('Name: {}, TotalNum:{}'.format(TotalLine[0], TotalLine[1]) + '\n\n')
    # 關閉文件
    resu.close()