前言:

学习过程中经常需要进行数据处理,其中最常见的操作就是读取某个文件夹中的所有文件名称,并将这些文件名称按照读取顺序一行一行写入txt文本文件中。记录一下

一.核心代码

  1. os.listdir(filePath) # os.listdir方法返回一个列表对象,参数为
    文件夹的路径,可以是”相对路径“和”绝对路径“
    2.在向txt文本文件中写入信息时,有时需要换行写入
    ==rowInfo = imageDir + ’ ’ + annotationDir == 不会换行写入
    ==rowInfo = imageDir + ’ ’ + annotationDir + ‘\n’ == 加上‘\n’,换行写入
import os
# readInfo函数,根据文件夹路径读取文件夹下所有文件名
def readInfo():
    filePath = 'F:\\ai\\PaddleSeg-release-2.4\\data\\island\\JPEGImages\\'
    name = os.listdir(filePath)         # os.listdir方法返回一个列表对象
    return name

# 程序入口
if __name__ == "__main__":
    fileList = readInfo()       # 读取文件夹下所有的文件名,返回一个列表
    print(fileList)
    file = open('train.txt', 'w')   # 创建文件,权限为写入
    for i in fileList:
        imageDir = 'JPEGImages/' + i
        annotationDir = 'Annotations/' + i
        rowInfo = imageDir + ' ' + annotationDir + '\n'
        print(rowInfo)
        file.write(rowInfo)

更多推荐

【python基础】python读取文件夹中文件名,并向txt文件中写入信息