우분투 폴더내 파일 랜덤 샘플링하기

import os
import random
import shutil

source = '../../dataset/dongkyun/dk_dataset/afhq_dataset/afhq/train/cat/cat_all'
dest1 = '../../dataset/dongkyun/dk_dataset/afhq_dataset/afhq/train/cat/cat_4739'


files = os.listdir(source)
no_of_files = 4739

print(no_of_files)
if not os.path.isdir(dest1):
    os.makedirs(dest1)
    
#import pdb;pdb.set_trace()
for file_name in random.sample(files, no_of_files):
    #move
    #shutil.move(os.path.join(source, file_name), dest1)
    #copy
    shutil.copy(os.path.join(source, file_name), os.path.join(dest1, file_name))