python实现从文件夹中随机选取一定数量的文件
# # -*- coding:utf-8 -*-# """# 作者:sunli# 日期:2022年03月14日14:06# """import osimport randomimport shutildef moveFile(fileDir, trainDir):pathDir = os.listdir(fileDir)# 取图片的原始路径filenumber = len(pathDir)rate
·
# # -*- coding:utf-8 -*-
# """
# 作者:sunli
# 日期:2022年03月14日14:06
# """
import os
import random
import shutil
fileDir = 'D:/Fall_dataset/MobiAct_Dataset_v2.0/Annotated Data'
trainDir = 'D:/Fall_dataset/MobiAct_Dataset_v2.0/train'
def moveFile(fileDir, trainDir):
pathDir = os.listdir(fileDir) # 取图片的原始路径
filenumber = len(pathDir)
# rate1 = 0.8 # 自定义抽取csv文件的比例,比方说100张抽80个,那就是0.8
rate1 = 0.8
picknumber1 = int(filenumber*rate1) # 按照rate比例从文件夹中取一定数量的文件
sample1 = random.sample(pathDir, picknumber1) # 随机选取picknumber数量的样本
for name in sample1:
shutil.move(fileDir + '/' + name, trainDir + "\\" + name)
if __name__ == '__main__':
file = os.listdir(fileDir)
for i in file:
moveFile(fileDir + '/' + i, trainDir)
# # -*- coding:utf-8 -*-
# """
# 作者:sunli
# 日期:2022年03月14日14:06
# """
import os
import random
import shutil
def moveFile(fileDir, trainDir):
pathDir = os.listdir(fileDir) # 取图片的原始路径
filenumber = len(pathDir)
rate1 = 0.8 # 自定义抽取csv文件的比例,比方说100张抽80张,那就是0.8
picknumber1 = int(filenumber * rate1) # 按照rate比例从文件夹中取一定数量的文件
sample1 = random.sample(pathDir, picknumber1) # 随机选取picknumber数量的样本
print(sample1)
for name in sample1:
shutil.move(fileDir + name, trainDir + "\\" + name)
# for i in pathDir:
# for j in sample1:
# if(i!=j):
# shutil.move(fileDir + name, testDir + "\\" + name)
#shutil.copyfile(fileDir + name, tarDir + name)
return
if __name__ == '__main__':
fileDir = "D:/sy/WAL/"
trainDir = 'D:/sy/WALtrain/'
#testDir ='D/sy/BSCtest'
moveFile(fileDir, trainDir)
更多推荐
已为社区贡献1条内容
所有评论(0)