python实现从文件夹中随机选取一定数量的文件

# # -*- 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)