如何将每 500 个文件移动到不同的文件夹中

Question

As a beginner in Python I would need your help since I do not know enough how to create such script for my need. To give you an overall idea I have a folder Folder_1 that contains 50 000 different frames from a video in .png :

Folder_1 :
    picture_00001
    picture_00002
    picture_00003
    picture_00004
    ...
    picture_50000

Since my Windows explorer is not running quite well with this huge amount of pictures I will need to move all of them in different folders in order to lower my RAM consumption and letting me working on a batch without considering the 50 000 pictures.

Therefore my objective is to have a script that will simply move the first 500 files to a folder sub_folder1 and then moving the 500 next to sub_folder2 etc... The folders needs to be created with the script as well :

Folder_1
    sub_folder1
         picture_00001
         picture_00002
         ...
         picture_00500

    sub_folder2
         picture_00501
         picture_00502
         ...
         picture_01000

I started working on with for i in range(500) but I have not clue on what to write then.

Hopping this is clear enough otherwise let me know and I will do my best to be even more precised.

Thank you in advance for your help.

Answer 1

One possible solution is:

First you find out which are the .png file names in the directory. You can achieve this by using os.listdir(<dir>) to return a list of filenames, then iterate over it and select just the correct files with fnmatch .

Then you set the increment (in this example 10 , in yours 500 ), iterate over a generator range(0, len(files), increment) , create a folder just if it doesn't exist and then move the files in chunks.

Solution:

from fnmatch import fnmatch
import os
import shutil

def get_filenames(root_dir):
    pattern = '*.png'
    files = []
    for file in os.listdir(root_dir):
        if fnmatch(file, pattern):
            files.append(file)
    return files

def distribute_files():
    root_dir = r"C:\frames"
    files = get_filenames(root_dir)
    increment = 10

    for i in range(0, len(files), increment):
        subfolder = "files_{}_{}".format(i + 1, i + increment)
        new_dir = os.path.join(root_dir, subfolder)
        if not os.path.exists(new_dir):
            os.makedirs(new_dir)
        for file in files[i:i + increment]:
            file_path = os.path.join(root_dir, file)
            shutil.move(file_path, new_dir)


if __name__ == "__main__":
    distribute_files()

Hope it helps.

Regards