Python 读取文件下所有内容、获取文件名、截取字符、写回文件

# coding=gbk

import os

import os.path

#读取目录下的所有文件，包括嵌套的文件夹

def GetFileList(dir, fileList):

    newDir = dir

    if os.path.isfile(dir):

        fileList.append(dir)

    elif os.path.isdir(dir):

        for s in os.listdir(dir):

            # 如果需要忽略某些文件夹，使用以下代码

            # if s == "xxx":

            # continue

            newDir = os.path.join(dir, s)

            GetFileList(newDir, fileList)

    return fileList

fileDir = "E:\Differnernt_Size_Digit_Data\ReSize\Train\28x28"

list = GetFileList(fileDir, [])

# 打开一个文件

fo = open("file_list.txt", "w")  # 打开文件

for i in list:

    print(i)  # 测试完整文件路径

    print(os.path.basename(i))  # 文件名

    index = i.find(".", 0)  # 找到点号的位置

    print(i[index - 1:index])  # 截取目标字符

    print(os.path.basename(i) + " " + i[index - 1:index])  # 测试目标字符串

    fo.write(os.path.basename(i) + " " + i[index - 1:index] + "
")  # 将目标字符串写入文件

fo.close()  # 关闭打开的文件