python+os根据文件名自动生成文本

yipeiwu_com6年前Python基础

有时我们有很多文件(如图片),我们需要对每一个文件进行操作。

我们还需要一份文件的名字来进行遍历,这时我们首先需要建立一份文件名单,有时还会对文件名做一定的筛选,如我们只选择jpg格式的图片。

本文就是利用一个文件夹下的几张图片完成这些操作:把所有文件名写入到一个文件中。

这可以进一步用来遍历处理数据

#! usr/bin/env python
# coding: utf-8

# 为了自动获取文件夹下的名字并将其写入文件,然后利用该文件来处理对应的文件


import os
import numpy as np
import cv2


# 先进行路径操作
currentRoot = os.getcwd() # 当前路径
imgRoot = currentRoot + '/img' # 图片所在的路径
os.chdir(imgRoot) # 切换到图片所在的路径


# 开始读取文件并将其名称写入一个文件
fileNames = open('imgName.txt','w') # 将要写入文件名的文件, w表示write
file = os.listdir(imgRoot) # 读取某个文件夹下的所有文件
for i in file: #采用文件句柄的方式遍历文件
 if os.path.splitext(i)[1]=='.jpg': # 判断是否是jpg文件,我们只对jpg文件进行操作
 data = str(i) + '\n' # 将文件名和换行放在一起,文件名采用相对路径
 fileNames.write(data) # 开始写入文件
fileNames.close() # 最后关掉文件

运行代码,可得到一个文件,可看下图

文件内容如下:

如果重新运行一次代码,发现文件内容不会变化。
其实这时候文件夹中已经有了imgName这个文件,但是其扩展名为txt,所以并不会被利用。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

怎样使用Python脚本日志功能

怎样使用Python脚本日志功能

  假设要开发一个自动化脚本工具,工程结构如下,Common这个package是框架功能的实现,Scripts目录是我们编写的测试用例脚本(请忽略其他不相关的目录)。   我们对日志功能...

Selenium+Python 自动化操控登录界面实例(有简单验证码图片校验)

Selenium+Python 自动化操控登录界面实例(有简单验证码图片校验)

从最简单的Web浏览器的登录界面开始,登录界面如下: 进行Web页面自动化测试,对页面上的元素进行定位和操作是核心。而操作又是以定位为前提的,因此,对页面元素的定位是进行自动化测试的基...

python操作小程序云数据库实现简单的增删改查功能

python操作小程序云数据库实现简单的增删改查功能

不止python,你可以利用任何语言那实现通过http请求来操作你自己的小程序云数据库了 背景 也是在最近吧,小程序更新了云开发 HTTP API 文档,提供了小程序外访问云开发资源的能...

Python利用全连接神经网络求解MNIST问题详解

Python利用全连接神经网络求解MNIST问题详解

本文实例讲述了Python利用全连接神经网络求解MNIST问题。分享给大家供大家参考,具体如下: 1、单隐藏层神经网络 人类的神经元在树突接受刺激信息后,经过细胞体处理,判断如果达到阈值...

Python中使用中文的方法

先来看看python的版本: >>> import sys >>> sys.version '2.5.1 (r251:54863, Apr...