Python处理文件的6个常用代码,使用频率很高,值得收藏
日常工作中,我们经常会遇到一些查找、存储文件的问题,比如文件合并、文件分解等,这些问题涉及到对文件进行操作处理。对此,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
也可从分离文件名后的列表中提取: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
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:
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、运行代码完成,此时汇总文件内已经合并了其他两个文件的数据。