python 随机抽取数据

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
'''
@author: carry
@contact: [email protected]
@file: move_copy.py
@time: 2019/4/8 9:16
@desc:
'''

import os, random, shutil


def moveFile(fileDir):
    pathDir = os.listdir(fileDir)  # 取图片的原始路径
    filenumber = len(pathDir)
    rate = 0.1  # 自定义抽取图片的比例，比方说100张抽10张，那就是0.1
    picknumber = int(filenumber * rate)  # 按照rate比例从文件夹中取一定数量图片
    sample = random.sample(pathDir, 1000)  # 随机选取picknumber数量的样本图片
    print(sample)
    for name in sample:
        shutil.move(fileDir + name, tarDir + name)
    return


if __name__ == '__main__':
    fileDir = "./trains/"  # 源图片文件夹路径
    tarDir = './predict/'  # 移动到新的文件夹路径
    moveFile(fileDir)

'''
# 复制文件：
shutil.copyfile("oldfile", "newfile")  # oldfile和newfile都只能是文件
shutil.copy("oldfile", "newfile")  # oldfile只能是文件夹，newfile可以是文件，也可以是目标目录

# 复制文件夹：
shutil.copytree("olddir", "newdir")  # olddir和newdir都只能是目录，且newdir必须不存在

# 重命名文件（目录）
os.rename("oldname", "newname")  # 文件或目录都是使用这条命令

# 移动文件（目录）
shutil.move("oldpos", "newpos")
shutil.move("D:/test.pdf", "D:/past/")
'''