一、使用Python批量创建folder
主要用到的库就是os；
代码运行的结果是：在指定文件夹下创建一组文件夹。
part1：代码：
import os #导入os模块
for i in range(1,11): #使用for循环创建从1到x的文件夹,此处是创建10个文件夹，从 1-10
path1 = 'D:/Codedata/test/creat_folder/' #设置创建后文件夹存放的位置,此处是在creat_folder文件夹下创建一组文件夹
path2 = '测试_' + str(i) #此处可以修改/删除：'测试_'
path = os.path.join(path1, path2) #路径拼接
isExist = os.path.exists(path) #定义一个变量判断文件是否存在
if not isExist: #如果文件不存在,则创建文件夹，并提示创建成功
os.makedirs(path)
print("%s 目录创建成功"%i)
else: #如果文件存在，则提示已经存在
print("%s 目录已经存在"%i)
continue #继续上述操作,直到循环结束
part2：运行结果图：
删掉第5行的 ‘测试_’ 后，path2 = str(i) ，意为只创建纯数字命名的文件夹，运行结果如下：
二、使用python批量创建txt
主要用到的库是os；
代码运行的结果是：在指定文件夹下创建一组txt。
part1：代码：
import os #导入os模块
for i in range(1,6): #使用for循环创建5个txt，从1到5
path1 = 'D:/Codedata/test/creat_folder/1/' #设置创建后文件夹存放的位置,此处是在creat_folder文件夹下的1文件夹下创建一组txt
path2 = '测试_' + str(i) #此处可以修改/删除：'测试_'
path = os.path.join(path1,path2) #路径拼接
f = open( path + '.txt',"a") #a表示没有该文件就新建
f.write("") #写入文件，空
f.close() #执行完结束
part2：运行结果图：
删掉第5行的 ‘测试_’ 后，path2 = str(i) ，意为只创建纯数字命名的txt，运行结果如下：
三、使用 Python批量修改文件名
主要用到的库os；
代码的作用是：可以修改指定文件夹下的文件，不限格式（包括：folder、txt等）。
part1：代码：
import os #导入os模块
path = 'D:/Codedata/test/creat_folder/' #需要修改的文件所在的路径
#此处是修改在creat_folder文件夹下的一组文件夹的名字
original_name = os.listdir(path) #读取文件初始的名字
print(original_name)
for i in original_name: #遍历全部文件
os.rename(os.path.join(path,i),os.path.join(path,'测试_'+i)) #修改文件名
part2：运行结果图：
四、使用Python读取txt中的数据并创建excel
首先，在第二节创建的txt的基础上，给每个txt写入一列9个数据，如下图所示：
接下来，先实现用Python读取一个txt的数据，并存入excel中，下面的代码针对“测试_1.txt”。
part1：代码：
## 本例中要读取的txt存放的路径为： D:/Codedata/test/creat_folder/1/
## 名字为： 测试_x.txt
import xlwt #导入excel文件的库
# 打开txt文件并读取数据
with open(r'D:/Codedata/test/creat_folder/1/测试_1.txt','r+') as title: #'r+'表示对文件是进行"读取和写入的模式"
layertitle = title.read()
layertitle_list= layertitle.split() #读取txt文件内容默认是str类型，此处将其分割成一个个元素形成列表
# 创建excel用于存放数据
workexcel = xlwt.Workbook(encoding='ascii') #创建一个工作表对象
sheet = workexcel.add_sheet('layer1',cell_overwrite_ok=True) #创建表格，添加一个名为layer1的sheet
i=0 #行变量，初始值为第一行
j=0 #列变量，初始值为第一列
for data in layertitle_list: #遍历数据，写入excel文件
sheet.write(i,j,data) #i，j控制表格坐标，左顶点为（0，0）
i=i+1 #控制j=0不变，i每次+1下移一格，表示将数据写成一列
workexcel.save('测试.xls') #保存文件，这里默认保存到相对路径，即当前工作区的路径
part2：运行结果图：
继续实现读取同一文件夹下的多个txt的数据，并存入一个表格的同一个sheet中。
思路：将上一步的代码封装成一个函数，只传入一个 txt_id 用于读取指定的.txt文件。
part1：代码：
#本例中要读取的txt存放的路径为： D:/Codedata/test/creat_folder/1/ 名字为： 测试_x.txt
import xlwt
import os
def read_txt(txt_id):
path1 = 'D:'
path2 = '//Codedata'
path3 = 'test'
path4 = 'creat_folder'
path5 = '1'
txt_id = '测试_' + str(txt_id) + '.txt'
path6 = txt_id
position = os.path.join(path1,path2,path3,path4,path5,path6)
print(position)
with open(position,'r+') as title: # 打开txt文件并读取数据
layertitle = title.read()
layertitle_list = layertitle.split()
return layertitle_list
if __name__ == '__main__':
workexcel = xlwt.Workbook(encoding='ascii') #创建一个工作表对象
sheet = workexcel.add_sheet('layer1',cell_overwrite_ok=True) #创建表格，添加一个名为layer1的sheet
for num in range(1,6): #利用for循环遍历txt文件
i=0 ##行变量，初始值为第一行
data = read_txt(num)
#print(data)
for d in data:
sheet.write(i,num,d) #i，j控制表格坐标，左定格为（0，1）下一次（0,2)以此类推
i=i+1 #按列写入数据
workexcel.save('测试2.xls') #保存文件，这里默认保存到相对路径
part2：运行结果图：
如果想实现读取不同文件夹下的txt数据，并存入同一表格的不同sheet怎么办？
思路：在上一步的read_txt函数中再引入一个folder_id，作为每个sheet的名字同时又能用于for循环中遍历所有文件夹。
part1：代码：
#本例中要读取的txt存放的路径为： D:/Codedata/test/creat_folder/x/ 名字为： 测试_x.txt
import xlwt
import os
def read_txt(txt_id,folder_id):
path1 = 'D:'
path2 = '//Codedata'
path3 = 'test'
path4 = 'creat_folder'
path5 = str(folder_id)
txt_id = '测试_' + str(txt_id) + '.txt'
path6 = txt_id
position = os.path.join(path1,path2,path3,path4,path5,path6)
print(position)
with open(position,'r+') as title: # 打开txt文件并读取数据
layertitle = title.read()
layertitle_list = layertitle.split()
return layertitle_list
if __name__ == '__main__':
workexcel = xlwt.Workbook(encoding='ascii') #创建一个工作表对象
for num1 in range(1,11): #利用for循环遍历文件夹
sheet = workexcel.add_sheet(str(num1),cell_overwrite_ok=True) #创建10个sheet
for num2 in range(1,6): #利用for循环遍历txt文件
i=0 ##行变量，初始值为第一行
data = read_txt(num2,num1)
#print(data)
for d in data:
sheet.write(i,num2,d) #i，j控制表格坐标，左定格为（0，1）下一次（0,2)以此类推
i=i+1 #按列写入数据
workexcel.save('测试3.xls') #保存文件，这里默认保存到相对路径
part2：运行结果图：
五、使用Python批量删除文件的第一行
面临的问题：自己创建的标签文件，打开之后发现第一行有version的信息，所以想把所有的标签的第一行都删掉~
主要用到的库就是os；
代码运行的结果是：对指定文件夹下的所有文件的第一行都进行删除的操作。
import os
for f in os.listdir('.'):
if '.xml' in f: #此处的.xml可以修改为.txt等文件格式
lines = open(f,encoding='utf-8').readlines()
open(f, 'w',encoding='utf-8').writelines(lines[1:])
part2：运行结果图：
六、使用Python批量修改xml文件的内容
面临问题：在处理一个数据集的标签时，我发现读取的分类标签name是同时含有大小写的，所以想把所有的分类都替换为小写的~
运行下面的代码时，需要注意的是区分：当前工作路径 和 xml文件存储的路径
part1：代码：
#第一步：导入库
import os
import os.path
from xml.etree.ElementTree import parse, Element
#第二步：定义一个函数，功能：批量修改xml中内容
def test():
path = "E://annotation" #xml文件所在的目录，举例 E:/annotation
files = os.listdir(path) # 获得文件夹下所有文件名
s = []
for xmlFile in files: # 遍历文件夹
if not os.path.isdir(xmlFile): # 判断是否是文件夹,不是文件夹才打开
print
xmlFile
pass
path = "E://annotation"
print(xmlFile) #输出文件名，方便知道在处理哪个xml文件
path1 = xmlFile #定位当前处理的文件的工作路径
newStr = os.path.join(path, xmlFile)
class1 = "bus" #这是你想最终显示的内容
class2 = "truck" #此处我想换成把原来的全部换成小写
class3 = "suv" #这里举了3个例子
dom = parse(newStr) #路径拼接,输入的是具体路径
root = dom.getroot()
#print(root)
for obj in root.iter('object'): #获取object节点中的name子节点
if obj.find('name').text == 'Bus': #判断获取的name节点下的内容是否为Bus
print('1') # 输出1
obj.find('name').text=class1
name1 = obj.find('name').text #修改
print(name1)
elif obj.find('name').text == 'Truck':
print('2')
obj.find('name').text=class2
name2 = obj.find('name').text #修改
print(name2)
elif obj.find('name').text == 'SUV':
print('3')
obj.find('name').text=class3
name3 = obj.find('name').text #修改
print(name3)
dom.write(path1, xml_declaration=True) #将修改后的xml保存到当前工作的路径下
pass
if __name__ == '__main__':
test()
part2：运行结果图：
七、拆分txt数据：一个txt分组成多个txt（选读Matlab）
【注】拆分一个包含很多数据的txt文件，使用的工具是matlab。
基础版：将一个包含73728（576*128）个数据的按列存放的txt文件，拆分成128组，每组的576个数据存于一个txt中。也就是说，代码运行之后，在指定文件夹下应该有128个txt，每个txt中有576个数据是按列存放的。
part1：代码：
A =importdata('D:/Codedata/test/newspace/forexcel/weights_32.txt'); %导入包含带分组数据的txt文件
B=reshape(A,576,128); %将数据分成128组，每组576个
mkdir('D:/Codedata/test/newspace/forexcel/','分组1'); %创建一个名为‘分组1’的文件夹
for j=1:128 %生成128个txt，每个txt中包含576个数据
n = strcat('D:/Codedata/test/newspace/forexcel/','分组1','/',num2str(j),'.txt');
fid=fopen(n,'w');
fprintf(fid,'%g/n',B(:,j));
fclose(fid);
end
part2：运行结果图：
打开1.txt 的结果–>
升级版：将上一步的128个txt进行拆分，每个txt对应一个文件夹，每一个文件夹下将576（9*64）个数据拆分成64组，每组的9个数据存于一个txt中。
part1：代码：
for i=1:128
mkdir('D:/Codedata/test/newspace/forexcel/分组2/',int2str(i)); %创建128个文件夹，同时将matlab的当前工作路径设置为此
filename = ['D:/Codedata/test/newspace/forexcel/分组1/',int2str(i),'.txt'];
A = load(filename); %使用load()函数读取128个txt文件的数据
B=reshape(A,9,64); %将数据分成64组，每组9个数据
for j=1:64 %生成64个txt，每个txt中包含9个数据
n = strcat('D:/Codedata/test/newspace/forexcel/分组2/',int2str(i),'/','测试_',num2str(j),'.txt');
fid=fopen(n,'w');
fprintf(fid,'%g/n',B(:,j));
fclose(fid);
end
end
part2：运行结果图：
八、读取.mat文件的数据，批量写入xml文件（选读Matlab）
面临问题：下载了一个网上的数据集，没有提供直接的标签，只给了一个.mat格式的文件，用Matlab读取，批量转换为符合VOC数据集格式的标签文件。
part1：代码：
data=load('D:/Program Files/MATLAB2019b/MYcode/VehicleInfo.mat') %.mat文件存储的路径
cars=data.VehicleInfo;
for n=1:length(cars)
n;
car=cars(n);
%%
%每一辆车新建一个xml文件存储车的信息
carName=car.name;
%图片的高度
carHeight=car.height;
%图片的宽度
carWidth=car.width;
%新建xml文件
annotation = com.mathworks.xml.XMLUtils.createDocument('annotation');
annotationRoot = annotation.getDocumentElement;
%定义子节点.xml的存储路径
folder=annotation.createElement('folder');
folder.appendChild(annotation.createTextNode(sprintf('%s','JPEGImages')));%这里为xml存放的目录
annotationRoot.appendChild(folder);
%图片的名称
jpgName=annotation.createElement('filename');
jpgName.appendChild(annotation.createTextNode(sprintf('%s',carName(end-10:end-4),'.jpg'))); %获取倒数第11-倒数第5的字符
annotationRoot.appendChild(jpgName);
%路径
jpgPath=annotation.createElement('path');
jpgPath.appendChild(annotation.createTextNode(sprintf('%s','/home/phm/E/YOLOX/YOLOX-main/datasets/JPEGImages/',carName(end-10:end-4),'.jpg')));
annotationRoot.appendChild(jpgPath);
%source
jpgSource=annotation.createElement('source');
annotationRoot.appendChild(jpgSource);
database=annotation.createElement('database');
database.appendChild(annotation.createTextNode(sprintf('%s','Unknown')));
jpgSource.appendChild(database);
%添加图片的size
jpgSize=annotation.createElement('size');
annotationRoot.appendChild(jpgSize);
%定义size的子节点
%图片宽度
width=annotation.createElement('width');
width.appendChild(annotation.createTextNode(sprintf('%i',carWidth)));
jpgSize.appendChild(width);
%图片高度
height=annotation.createElement('height');
height.appendChild(annotation.createTextNode(sprintf('%i',carHeight)));
jpgSize.appendChild(height);
%图片深度，彩色图片3
depth=annotation.createElement('depth');
depth.appendChild(annotation.createTextNode(sprintf('%i',3)));
jpgSize.appendChild(depth);
segmented=annotation.createElement('segmented');
segmented.appendChild(annotation.createTextNode(sprintf('%i',0)));%表示已经标注过了
annotationRoot.appendChild(segmented);
%接下来是每一辆车的标注信息
%%
carVehicles=car.vehicles;
L=length(carVehicles);
if L>1
carName
L
end
for nn=1:length(carVehicles)
vehicle=carVehicles(nn);
%标注框的最左侧坐标
vLeft=vehicle.left;
%标注框的最上边坐标
vTop=vehicle.top;
%标注框的最右侧坐标
vRight=vehicle.right;
%标注框最下面坐标
vBottom=vehicle.bottom;
%车的类别
vCategory=vehicle.category;
%图片中有多少个标注对象
carNvehicles=car.nVehicles;
%在这里生成每一符图片的txt文件，用于
%%注意一张图片中可能会有多个对象
%%matlab直接生成xml对象？？？
%将这一辆车的信息注入xml中
object=annotation.createElement('object');
annotationRoot.appendChild(object);
%标注框类别名称
categoryName=annotation.createElement('name');
categoryName.appendChild(annotation.createTextNode(sprintf('%s',vCategory)));
object.appendChild(categoryName);
pose=annotation.createElement('pose');
pose.appendChild(annotation.createTextNode(sprintf('%s','Unspecified')));
object.appendChild(pose);
truncated=annotation.createElement('truncated');
truncated.appendChild(annotation.createTextNode(sprintf('%i',0)));
object.appendChild(truncated);
difficult=annotation.createElement('difficult');
Difficult.appendChild(annotation.createTextNode(sprintf('%i',0)));
object.appendChild(Difficult);
bndbox=annotation.createElement('bndbox');
object.appendChild(bndbox);
xmin=annotation.createElement('xmin');
xmin.appendChild(annotation.createTextNode(sprintf('%i',vLeft)));
bndbox.appendChild(xmin);
ymin=annotation.createElement('ymin');
ymin.appendChild(annotation.createTextNode(sprintf('%i',vTop)));
bndbox.appendChild(ymin);
xmax=annotation.createElement('xmax');
xmax.appendChild(annotation.createTextNode(sprintf('%i',vRight)));
bndbox.appendChild(xmax);
ymax=annotation.createElement('ymax');
ymax.appendChild(annotation.createTextNode(sprintf('%i',vBottom)));
bndbox.appendChild(ymax);
end
%存储xml
savePath=['D:/BITVehicle_xml/',carName(1:end-3),'xml']; %指定文件夹用于存放生成的xml文件
xmlwrite(savePath,annotation);
end
九、使用Python读取某文件夹下所有文件并写入txt中
面临问题：在准备好数据集后，都是通过读取.txt的内容实现对图片的调用训练的，因此需要将一个包含图片的文件夹中的全部.jpg名字读成一个txt。
举例：以下文件夹是存放图片的文件夹，我想将它包含的10张图片读成一个.txt，详见如下两图介绍。
way1：生成的txt中只有图片名的代码：
需要先创建一个.txt用于存放信息，因为代码中没有创建的过程
注意修改自己的路径，一共有两处
#导入需要的库
import sys
import os
import random #导入随机函数
#存放原始图片地址
data_base_dir = r"D:/Codedata/test_pic"
#建立列表，用于保存图片信息
file_list = []
#读取图片文件，并将图片名写到txt文件中
write_file_name = 'D:/Codedata/test_pic.txt'
write_file = open(write_file_name, "w") #以只写方式打开write_file_name文件
for file in os.listdir(data_base_dir): #file为current_dir当前目录下图片名
if file.endswith(".jpg"): #如果file以jpg结尾
write_name = file #图片路径 + 图片名 + 标签
file_list.append(write_name) #将write_name添加到file_list列表最后
sorted(file_list) #将列表中所有元素随机排列
number_of_lines = len(file_list) #列表中元素个数
#将图片信息写入txt文件中，逐行写入
for current_line in range(number_of_lines):
write_file.write(file_list[current_line] + '/n')
#关闭文件
write_file.close()
way2：生成的txt包含图片路径及图片名的代码：
需要先创建一个.txt用于存放信息，因为代码中没有创建的过程
注意修改自己的路径，一共有三处
#导入需要的库
import sys
import os
import random #导入随机函数
#存放原始图片地址
data_base_dir = r"D:/Codedata/test_pic"
#建立列表，用于保存图片信息
file_list = []
#读取图片文件，并将图片地址、图片名写到txt文件中
write_file_name = 'D:/Codedata/test_pic.txt'
#以只写方式打开write_file_name文件
write_file = open(write_file_name, "w")
for file in os.listdir(data_base_dir): #file为current_dir当前目录下图片名
if file.endswith(".jpg"): #如果file以jpg结尾
write_name = file #图片路径 + 图片名 + 标签
file_list.append(write_name) #将write_name添加到file_list列表最后
sorted(file_list) #将列表中所有元素随机排列
number_of_lines = len(file_list) #列表中元素个数
#将图片信息写入txt文件中，逐行写入
for current_line in range(number_of_lines):
write_file.write('D://Codedata//test_pic//' + file_list[current_line] +'/n')
#关闭文件
write_file.close()
way3：如果是linux系统，在终端输入一条指令就可以生成txt，包含图片路径及图片名：
在想要生成txt的路径下打开终端，输入：
ls -R 包含图片的路径/* >xxx.txt
1
例如：我的包含图片的路径是/home/phm/F/fxh/test_resolution/picture/car_480x854_L1
我想在这个路径下生成一个input.txt，所以在此路径下打开终端，输入：
ls -R /home/phm/F/fxh/test_resolution/picture/car_480x854_L1/* > test.txt
生成的input.txt的内容如下：【注意：此方法仅适用于图片数量较少的情况】
十、使用Python根据已有的txt文件删除某文件夹下的文件
在上一节的基础上，我们已经获得了包含图片路径及图片名的.txt文件，现在想利用这个txt删除指定文件夹下的图片文件
先展示已经具备的文件环境：
test_pic的文件夹中有需要删掉的图片，也有需要保留的图片
可以通过test_pic.txt中包含的图片路径对此文件夹进行操作
假设想删除1-10.jpg，保留1-10-副本.jpg
part1：代码：
#导入库
import os
#把txt读成列表data
input_file = r'D:/Codedata/test_pic.txt' ##需要删除的文件名组成的txt文件
with open(input_file, 'r') as f:
data = f.read().splitlines()
#读取待删除的文件夹下的全部文件，读成列表filelist
rootdir=r'D:/Codedata/test_pic'
filelist=os.listdir(rootdir)
#删除操作
for file in data:
if '.jpg' in file:
del_file = file #使用txt中的路径
os.remove(del_file)#删除文件
print("已经删除：",del_file)
part2：运行结果图：
如果.txt是只包含图片名，而没有路径的，则需要在上述代码中做一处修改
第16行的del_file = file修改为：
del_file = rootdir + '//' + file #当代码和要删除的文件不在同一个文件夹时，必须使用绝对路径
原创文章，作者：ItWorker，如若转载，请注明出处：https://blog.ytso.com/281572.html