Python处理文件的常用代码

简介: Python处理文件的常用代码

日常工作中,我们经常会遇到一些查找、存储文件的问题,比如文件合并、文件分解等,这些问题涉及到对文件进行操作处理。对此,Python的OS库提供了很多功能模块供使用,本文整理了其中6个使用频率很高的常用代码,基本能满足大部分需求。另外,通过对一个excel文件合并案例代码进行解析,进一步加深印象,建议收藏。

一、6个常用代码
1、得到指定文件路径:os.path.dirname(带路径文件名)
查找处理文件要调用os库,和openpyxl一样,先用import 导入。

import os

file = r'e:\python\删除最小值.xlsx'

pwd = os.path.dirname(file)

print(pwd)

运行结果为:e:\python
//代码效果参考:http://www.ezhiqi.com/zx/art_4708.html

也可从分离文件名后的列表中提取:os.path.split(带路径文件名)[0]

2、得到当前文件名:os.path.dasename(带路径文件名)
import os

file = r'e:\python\删除最小值.xlsx'

pwdn = os.path.basename(file)

print(pwdn)

运行结果为:删除最小值.xlsx

也可从分离文件名后的列表中提取:os.path.split(带路径文件名)[1]

3、判断文件路径是否存在:os.path.exists(带路径文件名)
import os

file = r'e:\python\删除最小值.xlsx'

pwdbool = os.path.exists(file)

print(pwdbool)

运行结果为:true

若给定的路径文件不存在,则返回false

4、获得指定文件夹下所有文件:print(os.listdir(文件夹路径))
import os
//代码效果参考:http://www.ezhiqi.com/zx/art_1350.html

pwdns = os.listdir(r'e:\python')

print(pwdns[1])

运行结果:8.py 为'e:\python'文件夹下第2个文件(或文件夹)名。

5、获得文件后缀:os.path.splitext(文件名)
import os

pdn= os.path.splitext('试验.py')

print(pdn[1])

运行结果:.py,若最后一句改为print(pdn[0]),则得到文件名。

6、拼接路径和文件名:os.path.join(路径,文件名)
f2 = os.path.join(f1, '汇总.xlsx')

运行结果:'e:\python\汇总\汇总.xlsx'

二、文件合并案例代码解析:
在下面的文件夹“e:\python\汇总”里,有2个xlsx文件和一个txt文件,要求先判断当前文件夹内是否存在“汇总.xlsx”文件,若不存在则新建一个,若存在则直接打开使用,要求将其他.xlsx文件的第1个表单数据,汇总至“汇总.xlsx”文件。

初始文件

1、导入os库和openpyxl库,给定当前路径为模块参数

import os

import openpyxl

if name == "main":

tj(r'e:\python\汇总')# 指定文件夹作为参数,**文末有提醒事项"

2、在tj模块内写代码,先判断是否存在汇总.excle文件,若无则创建

def tj(f1):

f2 = os.path.join(f1, '汇总.xlsx')# 组合文件路径,**文末有提醒事项"

bl = os.path.exists(f2)# 判断'e:\python\汇总\汇总.xlsx'是否存在,存在返回true,不存在返回false

if bl is False:
//代码效果参考:http://www.ezhiqi.com/bx/art_851.html

wb = openpyxl.Workbook()# 如果不存在,创建新excel文件

ws = wb.active

ws.title = "汇总"# 设置当前表单名为“汇总”

wb.save(f2)# 将刚才创建的文件保存为:'e:\python\汇总\汇总.xlsx'

wb = openpyxl.load_workbook(f2)# 打开'e:\python\汇总\汇总.xlsx'文件

ws = wb['汇总']

ks = ws.max_row# 得到汇总表单的最大行,**文末有提醒事项"

3、遍历指定文件夹,将目标.xlsx文件放入列表k备用

pns = os.listdir(f1)# 获得f1文件夹下带后缀文件名列表

k = []

for i in range(len(pns)):

hz = os.path.splitext(pns[i])[1]# 获得f1文件夹下文件名列表

qz = os.path.splitext(pns[i])[0]# 获得f1文件夹下文件后缀列表

if hz == '.xlsx' and qz != '汇总':

k.append(pns[i])# 将后缀为.xlsx,且文件名不是汇总的文件名放入K列表

5、逐个打开目标文件,将第一个表单内的数据复制粘贴在汇总表内汇总表单内

m = 0

for w in k:

f3 = os.path.join(f1, w)

wb1 = openpyxl.load_workbook(f3)

ws1 = wb1._sheets[0]# 选第一个表单,默认数据在第一个表单,也可根据表单名进行选取。

for jj in range(1, ws1.max_row):

m += 1

for j in range(2, ws1.max_column + 1):

ws.cell(ks+m, j).value = ws1.cell(jj+1, j).value

ws.cell(ks+m, 1).value = ks+m-1# 重新写序号

ws.cell(ks+m, ws1.max_column+1).value = w# 写入当前行数据的来源文件名

wb.save(f2)# 保存文件

6、运行代码完成,此时汇总文件内已经合并了其他两个文件的数据。

程序运行后新建了汇总.xlsx

汇总文件内已完成数据汇总

三、几个细节
1、tj(r'e:\python\汇总'):给定参数时,路径前面有个字母“r”,为消除转义字符功能用。

2、os.path.join(f1, w):路径和文件名组合,也可用f1+'\'+w的方法,但只能限定于在打开文件的代码里用,单独使用会出现双“\”符号,导致地址无法使用。

3、ws1.max_row:得到表单的最大行数,当末行没有数据,只是设置过单元格格式,也被视为表单的最大行,这点不如VBA的end(3).row,需注意。

相关文章
|
22天前
|
存储 算法 调度
【复现】【遗传算法】考虑储能和可再生能源消纳责任制的售电公司购售电策略(Python代码实现)
【复现】【遗传算法】考虑储能和可再生能源消纳责任制的售电公司购售电策略(Python代码实现)
128 26
|
25天前
|
测试技术 开发者 Python
Python单元测试入门:3个核心断言方法,帮你快速定位代码bug
本文介绍Python单元测试基础,详解`unittest`框架中的三大核心断言方法:`assertEqual`验证值相等,`assertTrue`和`assertFalse`判断条件真假。通过实例演示其用法,帮助开发者自动化检测代码逻辑,提升测试效率与可靠性。
164 1
|
28天前
|
机器学习/深度学习 算法 调度
基于多动作深度强化学习的柔性车间调度研究(Python代码实现)
基于多动作深度强化学习的柔性车间调度研究(Python代码实现)
111 1
|
11天前
|
测试技术 Python
Python装饰器:为你的代码施展“魔法”
Python装饰器:为你的代码施展“魔法”
196 100
|
11天前
|
开发者 Python
Python列表推导式:一行代码的艺术与力量
Python列表推导式:一行代码的艺术与力量
202 95
|
19天前
|
Python
Python的简洁之道:5个让代码更优雅的技巧
Python的简洁之道:5个让代码更优雅的技巧
174 104
|
19天前
|
开发者 Python
Python神技:用列表推导式让你的代码更优雅
Python神技:用列表推导式让你的代码更优雅
304 99
|
11天前
|
缓存 Python
Python装饰器:为你的代码施展“魔法
Python装饰器:为你的代码施展“魔法
129 88
|
26天前
|
IDE 开发工具 开发者
Python类型注解:提升代码可读性与健壮性
Python类型注解:提升代码可读性与健壮性
198 102
|
16天前
|
监控 机器人 编译器
如何将python代码打包成exe文件---PyInstaller打包之神
PyInstaller可将Python程序打包为独立可执行文件,无需用户安装Python环境。它自动分析代码依赖,整合解释器、库及资源,支持一键生成exe,方便分发。使用pip安装后,通过简单命令即可完成打包,适合各类项目部署。

推荐镜像

更多