编程语言
首页 > 编程语言> > python 随机抽取数据

python 随机抽取数据

作者:互联网

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
'''
@author: carry
@contact: 864140438@qq.com
@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/")
'''

标签:__,抽取,name,python,move,fileDir,文件夹,随机,shutil
来源: https://blog.csdn.net/xkx_07_10/article/details/89097369