python随机选取目录下的若干个文件

最新推荐文章于 2023-06-20 14:43:33 发布

weixin_30878501

最新推荐文章于 2023-06-20 14:43:33 发布

阅读量2.6k

点赞数

文章标签： python java

原文链接：http://www.cnblogs.com/sdu20112013/p/11084151.html

版权

个人记录用. python模块random argparse shutil

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('num',type=int,help="img numbers to random")
args = parser.parse_args()

import random
import os
path="/home/train/disk/data/yulan_park_expand"
imgs = []
for x in os.listdir(path):
    if x.endswith('jpg'):
    imgs.append(x)
selected_imgs=random.sample(imgs,k=args.num)
print(selected_imgs)

from shutil import copyfile
for img in selected_imgs:
    src=os.path.join(path,img)
    dst=os.path.join(path,"../for_bitmain/"+img)    
    copyfile(src,dst)
print("copy done")

转载于:https://www.cnblogs.com/sdu20112013/p/11084151.html