Python中常见的文件格式处理

Python中常见的文件格式处理

1.读写EXCEL文件

python语言处理excel文件（xls、xlsx）时，一般会使用openyxl库和xlrd,xlwt库，这里推荐使用openyxl库进行处理，其处理的excel行数更大

1.1 EXCEL文件读取

from openpyxl import load_workbook
#读取的excel名称
excel = load_workbook("./test.xlsx") 
sheet = excel.get_sheet_by_name("Sheet1") #读取excel的sheet表格
rows = sheet.max_row      #获取行数
cols = sheet.max_column   #获取列数
for row in range(1,rows+1):
    id = sheet.cell(row = row,column=1).value     #第一列
    source = sheet.cell(row=row,column=2).value    #第二列
    target = sheet.cell(row=row,column=3).value    #第三列
    print(id,"->",target)
    
if __name__ == "__main__":
    print("excel读取完毕")

1.2 EXCEL文件书写

#!/usr/bin/python
# -*- coding: UTF-8 -*-
#Author: chengjinpei
import openpyxl
def write_xlsx_file(output_file):
    workbook = openpyxl.Workbook()                #创建excel对象
    worksheet = workbook.create_sheet("Sheet1")    #创建excel中的sheet
    i = 1
    for i in range(1,101):
        worksheet.cell(i,1,"flowid")
        worksheet.cell(i,2,"context")
    workbook.save(output_file)  #保存文件

if __name__ == "__main__":
    write_xlsx_file("test_output.xlsx")

2.txt文本的读写

python读取txt文本比较简单，直接使用open即可

input_file = "test.txt"
output_file = "test_out.txt"

fr = open(input_file,"r",encoding="utf-8")
fw = open(output_file,"w",encoding="utf-8")
lines = fr.readlines()
for line in lines:
    print(line)
    fw.write(line)
   #fw.flush()
if __name__ == "__main__":
    print("读写文件完毕")