Python的文件操作

一、图片操作

Train_path = 'D:/BaiduNetdiskDownload/数据集/分类/猫狗分类数据集/data/train'
Traincat_aimpath = 'data/train/cat'
Traindog_aimpath = 'data/train/dog'

Testcat_aimpath = 'data/test/cat'
Testdog_aimpath = 'data/test/dog'

1、读取某文件夹的图片

此时img_list存有Train_path文件夹中所有内容，是一个数组

Train_path = 'D:/BaiduNetdiskDownload/数据集/分类/猫狗分类数据集/data/train'   
img_list = os.listdir(Train_path )

遍历img_list，获取文件夹中的每一个图片名称。

    for img_name in img_list:
        # print(img_name )

img_name指的是图片的名称，示例：cat.1.jpg

名称字符串分割，获取cat和number

 name = img_name.split('.')[0]
        number = img_name.split('.')[1]
        number = int(number)

2、转移或者保存图片

将Train_path中的根据number转移3000张图片到目标文件夹中，需要先打开源图片，在进行save

   if number <= 3000:
            if name == 'cat':
                # aimfile_path：图片要迁移的路径名，
                aimfile_path = os.path.join(cataim_path, img_name)
                if not os.path.exists(aimfile_path):
                    # 先打开源路径的图片，然后再保存在目标路径
                    file_img = Image.open(Train_path + '/' + img_name)
                    file_img.save(aimfile_path)

3、删去某文件夹的图片

os.remove('data/test/cat/cat.3001.jpg')

4、显示相应图片

imgpath=os.path.join(train_path,img_name)
file_img = Image.open(train_path+'/'+img_name)
file_img.show()

 

二、文件夹操作

test_path = 'data/test'

1、生成文件夹

os.mkdir(test_path+'/'+'qw')