从一个文件夹读取文件，将符合要求的文件重写到另一个文件夹

import os
def listdir(path):
    for file in os.listdir(path):
        file_path = os.path.join(path, file)
        print("file_path====",file_path)
        file_path = file_path.replace('\\', '/')
        rewrite_file(file_path,file)
def rewrite_file(file_path,file):
    with open(file_path, mode='r+', encoding='utf-8') as f:
        with open("train2017/"+file, mode='w', encoding='utf-8') as ff:
            for line in f.readlines():
                print("line==", line)
                if line.split(' ')[0]=='0' in line:
                    ff.write(line)  # 重写数据
    f.close()
    ff.close()
def upper_data(data):
    return data.upper()

if __name__ == '__main__':

    listdir("train2017tmp")