python中常用的utils

菜菜码农一枚

已于 2022-11-29 20:52:41 修改

阅读量1.8k

点赞数

文章标签： python

于 2022-08-18 10:53:41 首次发布

本文链接：https://blog.csdn.net/Walter_fdh/article/details/126401336

版权

本文介绍了如何使用Python从input.txt文件中随机选择N行写入output.txt，并演示了如何从文件夹中随机选取样本并复制到dst_path。涉及到了文件读写、随机选择和目录操作的基础知识。

摘要由CSDN通过智能技术生成

逐行读写txt文件

从input.txt中逐行读取写为list，随机选择N个写入output.txt

import os
import numpy as np
import random

os.chdir(os.path.abspath(os.path.dirname(os.path.abspath(__file__)) + os.path.sep + "."))


N = 10
in_path = "./input.txt"
cur_path = os.getcwd()

out_path = os.path.join(cur_path, "output.txt")
lists = list()

f = open(in_path, "r")
lines = f.readlines()

for line in lines:
    line = line.strip("\n")
    lists.append(line)
f.close()

sele_lists = random.sample(lists, N)

with open(out_path, 'a+') as f:
    for file in sele_lists:
        f.writelines(file + "\n")
f.close()

文件复制

python中从一个文件夹中随机选择N个样本复制

import os
import shutil
import numpy as np

src_path = './src'
dst_path = './dst'
imgs = os.listdir(src_path)
sele_imgs = np.random.choice(imgs, 10000, replace=False)
lists = [os.path.join(src_path, x) for x in sele_imgs]
for img in lists:
    shutil.copy(img, dst_path)