python隨機選取10000張圖片並複製到另一個文件夾中

# author by LYS 2017/5/24
# for Deep Learning course
'''
1. read the whole files under a certain folder
2. chose 10000 files randomly
3. copy them to another folder and save
'''
import os, random, shutil


def copyFile(fileDir,tarDir):
    # 1
	pathDir = os.listdir(fileDir)

    # 2
	sample = random.sample(pathDir, 10000)
	print sample
	
	# 3
	for name in sample:
		shutil.copyfile(fileDir+name, tarDir+name)
if __name__ == '__main__':
	fileDir = "/home/lys/img_align_celeba_png/"
	tarDir = '/home/lys/celebA/'
	copyFile(fileDir,tarDir)
遇到的問題:
 shell命令行下for循環內部要有四個空格;回車兩次for循環執行;copy文件時,目錄最後要有/,否則直接把文件名字添加在了目錄後面,更改路徑
發佈了32 篇原創文章 · 獲贊 16 · 訪問量 7萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章