Python——文件处理

• os.listdir(路径)
  遍历文件和文件夹，把路径下的文件/文件夹转换成一个list
• os.path.join()
  扩展路径

import os
path = r"C:\Users\Administrator\Desktop"
filelist = os.listdir(path) #该文件夹下所有的文件（包括文件夹）

for file in filelist:
    print(file)

---

如果只想遍历文件，而需要跳过文件夹

import os
path = r"C:\Users\Administrator\Desktop"
filelist = os.listdir(path)

for file in filelist:
	Olddir = os.path.join(path,file)
	if os.path.isdir(Olddir):
		continue
	print(file)

• 批量文件移动

import os
import shutil


path = r'E:\graduation_project\object_detection-master\DCL-master\CUB_200_2011\CUB_200_2011\images'       # 想要查找的文件夹路径
new_path = r'E:\graduation_project\object_detection-master\DCL-master\datasets\CUB\data'            # 想要保存的文件夹路径，需新建一个文件夹

for root, dirs, files in os.walk(path):
    for i in range(len(files)):
        print(files[i])     # 打印出文件夹中所有的文件名，包括本文件夹中子文件夹中的文件
        # 匹配想要复制转移的文件类型
        if (files[i][-3:] == 'mp4') or (files[i][-3:] == 'png') or (files[i][-3:] == 'jpg'):
            file_path = root + '/' + files[i]
            new_file_path = new_path + '/' + files[i]
            shutil.copy(file_path, new_file_path)

yn_close = input('是否退出？')   # 回车退出

• 批量生成txt文件，并且写入内容
  如果没有该文件，open()会自动生成一个文件

path = r"C:\Users\Administrator\Desktop"
name = r'\train'
full_path = path + name + '.txt' 
file = open(full_path,'w')
file.write('hello')
file.close