Question

在每个子目录中都有不同的jpeg，png。

我要：

从这些子目录中选择X张随机图像

创建一个新文件夹并在其中复制这些选定的随机图像。

感谢已经在这里获得帮助，我可以使用os.walk和random.choice打印出随机选择的图像。

import os
import random
import shutil


files_list = []
for root, dirs, files in os.walk("/Path/to/Directory"):
    for file in files:
        #all 
        if file.endswith(".jpg") or file.endswith(".png") or file.endswith(".jpeg"):
            files_list.append(os.path.join(root, file))


#print images
#lets me count and print the amount of jpeg,jpg,pmg 
file_count = len(files_list)
print file_count

print files_list   
print(random.sample(files_list, 2))  #prints two random files from list

但是，我的问题是实际上选择随机图像（而不是它们的名称）

我尝试创建使用os.walk的变量imagePath

#creates a variable imagePath that lets me access all img files in different folders
imagePath = os.walk("/Path/to/Directory")

和一个新变量，可从imagePath

中随机选择单个图像

#create a variable that lets me choose random iamge from imagePath
randomImages = random.choice(os.listdir(imagePath))

，然后创建一个新目录，并使用shutil.copy将径向选择的图像移到该新目录中

#creates a new directory 
os.mkdir('testDirectory')

#moves the randomly selected image into new directory 
shutil.copy(randomImages, testDirectory)

但是，出现以下错误：

Traceback (most recent call last):
  File "crawl.py", line 28, in <module>
    randomImages = random.choice(os.listdir(imagePath)) 
TypeError: coercing to Unicode: need string or buffer, generator found

我也尝试过


for root, dirs, files in os.walk("/Path/to/Directory", topdown=False):
    imagePath = ("/Path/to/Directory") #creates a variable that lets me access all img files in different folders
    randomImages = random.choice(os.listdir(imagePath)) 
    print randomImages

但这会返回随机选择的子目录（而不是其中的图像）以及.ds存储文件。

Answer 1

您应该能够提供shutil.copy（）源和目标文件路径。在我看来，您已经有了文件列表，因此您只需复制它们即可。

import os
import random
import shutil


files_list = []
for root, dirs, files in os.walk("/Path/to/Directory"):
    for file in files:
        #all 
        if file.endswith(".jpg") or file.endswith(".png") or file.endswith(".jpeg"):
            files_list.append(os.path.join(root, file))


#print images
#lets me count and print the amount of jpeg,jpg,pmg 
file_count = len(files_list)
print file_count

print files_list   
selected_files = random.sample(files_list, 2))  #assign to a list

dest_path = "/path/to/new/folder/"
os.mkdir(dest_path)

for src_path in selected_files:
    shutil.copy(src_path, os.path.join(dest_path, os.path.basename(src_path)))

Answer 2

这是代码，我想移动文件而不是复制另一个文件。

import os
import random
import shutil


files_list = []

for root, dirs, files in os.walk("<SOURCE_DIR>"):
    for file in files:
        #all 
        if file.endswith(".jpg") or file.endswith(".png") or file.endswith(".jpeg"):
            files_list.append(os.path.join(root, file))


#print images
#lets me count and print the amount of jpeg,jpg,pmg 
file_count = len(files_list)
print file_count

# print files_list   
filesToCopy = random.sample(files_list, 2)  #prints two random files from list 

destPath = "<DESTINATION_DIR>"

# if destination dir does not exists, create it
if os.path.isdir(destPath) == False:
        os.makedirs(destPath)

# iteraate over all random files and move them
for file in filesToCopy:
    shutil.move(file, destPath)

将随机选择的图像放入新文件夹

2 个答案: