python - shutil 和复制文件

Question

我有一个很大的图像数据集(大约 3000 个文件)。我的问题很简单，我想将随机选择的图像文件复制到另一个目的地。我使用 random.sample 选择五百张图片并将它们的名称存储在列表中。 我现在想将文件从 src 文件夹复制到目标文件夹，如果它们的名称存在于列表中(因此是随机选择的)。

然而，以下代码复制文件夹中的所有文件，无论它们的名称是否出现在随机选择的列表中。帮助

import os.path
import os
import glob
import random
import shutil


dirfiles = os.listdir("/media/Data/Leaves/Leaves")
myfiles = []
myfiles.append(random.sample(dirfiles,500))


print myfiles


final_list=myfiles[0]
print final_list
count=0

for elem in final_list:
    print elem
    count= count+1

print count


src = '/home/mjanja/Desktop/Leaves'
dst = '/home/mjanja/Desktop/Positive Leaves'
for filename in final_list:
    for file in glob.glob( os.path.join(src,filename)):
        shutil.copy(file,dst)

print "Copied file!!" +infile

Answer 1

您正在做一些过于复杂的事情，例如计算元素和在不同位置定义目录。

这个怎么样？

import os
import random
import shutil

src = '/home/mjanja/Desktop/Leaves'
dst = '/home/mjanja/Desktop/Positive Leaves'

dirfiles = os.listdir(src)
myfiles = random.sample(dirfiles, 500)
count = len(myfiles)

for filename in myfiles:
    shutil.copy(os.path.join(src, filename), dst)