Python操作Excel數據的封裝函數分享
對比其它編程語言,我們都知道Python最大的優(yōu)勢是代碼簡單,有豐富的第三方開源庫供開發(fā)者使用。伴隨著近幾年數據分析的熱度,Python也成為最受歡迎的編程語言之一。而對于數據的讀取和存儲,對于普通人來講,除了數據庫之外,最常見的就是微軟的Excel。
1. 前期準備
1.1. 初識Excel
Microsoft Excel是Microsoft為使用Windows和Apple Macintosh操作系統(tǒng)的電腦編寫的一款電子表格軟件。
1.2. 格式區(qū)別
Excel中有xls和xlsx兩種格式,它們之間的區(qū)別是:
- 文件格式不同。xls是一個特有的二進制格式,其核心結構是復合文檔類型的結構,而xlsx的核心結構是XML類型的結構,采用的是基于 XML的壓縮方式,使其占用的空間更小。xlsx 中最后一個 x 的意義就在于此。
- 版本不同。xls是Excel2003及以前版本生成的文件格式,而xlsx是Excel2007及以后版本生成的文件格式。
- 兼容性不同。xlsx格式是向下兼容的,可兼容xls格式。
1.3. 庫的使用
Python自帶的模塊中有針對xls格式的xlrd和xlwt模塊,但這兩個庫僅僅是針對xls的操作,當我們要操作xlsx格式文件時,則需要使用到openpyxl第三方庫。
1.4. 整體思路
當使用以上幾個模塊的時候,從理論上我們就可以完全操作不同格式的Excel的讀和寫,很多人就疑惑,那這篇文章的作用是什么?我們直接學習對應的這三個模塊不就好了嗎?
答案就是:雖然這幾個庫已經把Excel的文件、表、行、列的概念完全轉換為Python中的對象,但每次操作都需要遍歷每一個單元格,甚至很多時候我們要花費大量的時間在思考循環(huán)單元格的邊界上,這本身就是在重復造輪子,因此我花了半天時間整理了以下六個函數。
2. 代碼展示
2.1. xlz格式
2.1.1. 讀取xls格式文件
def?read_xls_excel(url,index): ????''' ????讀取xls格式文件 ????參數: ????????url:文件路徑 ??????? index:工作表序號(第幾個工作表,傳入參數從1開始數) ????返回: ????????data:表格中的數據 ????''' ????#?打開指定的工作簿 ????workbook?=?xlrd.open_workbook(url) ????#?獲取工作簿中的所有表格 ????sheets?=?workbook.sheet_names() ????#?獲取工作簿中所有表格中的的第?index?個表格 ????worksheet?=?workbook.sheet_by_name(sheets[index-1]) ????#?定義列表存儲表格數據 ????data?=?[] ????#?遍歷每一行數據 ????for?i?in?range(0,?worksheet.nrows): ????????#?定義表格存儲每一行數據 ????????da?=?[] ????????#?遍歷每一列數據 ????????for?j?in?range(0,?worksheet.ncols): ????????????#?將行數據存儲到da列表 ????????????da.append(worksheet.cell_value(i,?j)) ????????#?存儲每一行數據 ????????data.append(da) ????#?返回數據 ????return?data
2.1.2. 寫入xls格式文件
def?write_xls_excel(url,sheet_name,two_dimensional_data): ??''' ????寫入xls格式文件 ????參數: ????????url:文件路徑 ????????sheet_name:表名 ??????? two_dimensional_data:將要寫入表格的數據(二維列表) ????''' ????#?創(chuàng)建工作簿對象 ????workbook?=?xlwt.Workbook() ????#?創(chuàng)建工作表對象 ????sheet?=?workbook.add_sheet(sheet_name) ????#?遍歷每一行數據 ????for?i?in?range(0,len(two_dimensional_data)): ????????#?遍歷每一列數據 ????????for?j?in?range(0,len(two_dimensional_data[i])): ????????????#?寫入數據 ????????????sheet.write(i,j,two_dimensional_data[i][j]) ????#?保存 ????workbook.save(url) ????print("寫入成功")
2.1.3. 追加寫入xls格式文件
def?write_xls_excel_add(url,?two_dimensional_data,?index): ????''' ????追加寫入xls格式文件 ????參數: ????????url:文件路徑 ??????? two_dimensional_data:將要寫入表格的數據(二維列表) ??????? index:指定要追加的表的序號(第幾個工作表,傳入參數從1開始數) ????''' ????#?打開指定的工作簿 ????workbook?=?xlrd.open_workbook(url) ????#?獲取工作簿中的所有表格 ????sheets?=?workbook.sheet_names() ????#?獲取指定的表 ????worksheet?=?workbook.sheet_by_name(sheets[index-1]) ????#?獲取表格中已存在的數據的行數 ????rows_old?=?worksheet.nrows ????#?將xlrd對象拷貝轉化為xlwt對象 ????new_workbook?=?copy(workbook) ????#?獲取轉化后工作簿中的第index個表格 ????new_worksheet?=?new_workbook.get_sheet(index-1) ????#?遍歷每一行數據 ????for?i?in?range(0,?len(two_dimensional_data)): ????????#?遍歷每一列數據 ????????for?j?in?range(0,?len(two_dimensional_data[i])): ????????????#?追加寫入數據,注意是從i+rows_old行開始寫入 ????????????new_worksheet.write(i+rows_old,?j,?two_dimensional_data[i][j]) ????#?保存工作簿 ????new_workbook.save(url) ????print("追加寫入成功")
2.2. xlsx格式
2.2.1. 讀取xlsx格式文件
def?read_xlsx_excel(url,?sheet_name): ????''' ????讀取xlsx格式文件 ????參數: ????????url:文件路徑 ????????sheet_name:表名 ????返回: ????????data:表格中的數據 ????''' ????#?使用openpyxl加載指定路徑的Excel文件并得到對應的workbook對象 ????workbook?=?openpyxl.load_workbook(url) ????#?根據指定表名獲取表格并得到對應的sheet對象 ????sheet?=?workbook[sheet_name] ????#?定義列表存儲表格數據 ????data?=?[] ????#?遍歷表格的每一行 ????for?row?in?sheet.rows: ????????#?定義表格存儲每一行數據 ????????da?=?[] ????????#?從每一行中遍歷每一個單元格 ????????for?cell?in?row: ????????????#?將行數據存儲到da列表 ????????????da.append(cell.value) ????????#?存儲每一行數據 ????????data.append(da) ????#?返回數據 ????return?data
2.2.2. 寫入xlsx格式文件
def?write_xlsx_excel(url,?sheet_name,?two_dimensional_data): ????''' ????寫入xlsx格式文件 ????參數: ????????url:文件路徑 ????????sheet_name:表名 ??????? two_dimensional_data:將要寫入表格的數據(二維列表) ????''' ????#?創(chuàng)建工作簿對象 ????workbook?=?openpyxl.Workbook() ????#?創(chuàng)建工作表對象 ????sheet?=?workbook.active ????#?設置該工作表的名字 ????sheet.title?=?sheet_name ????#?遍歷表格的每一行 ????for?i?in?range(0,?len(two_dimensional_data)): ????????#?遍歷表格的每一列 ????????for?j?in?range(0,?len(two_dimensional_data[i])): ????????????#?寫入數據(注意openpyxl的行和列是從1開始的,和我們平時的認知是一樣的) ????????????sheet.cell(row=i?+?1,?column=j?+?1,?value=str(two_dimensional_data[i][j])) ????#?保存到指定位置 ????workbook.save(url) ????print("寫入成功")
2.2.3. 追加寫入xlsx格式文件
def?write_xlsx_excel_add(url,?sheet_name,?two_dimensional_data): ????''' ????追加寫入xlsx格式文件 ????參數: ????????url:文件路徑 ????????sheet_name:表名 ??????? two_dimensional_data:將要寫入表格的數據(二維列表) ????''' ????#?使用openpyxl加載指定路徑的Excel文件并得到對應的workbook對象 ????workbook?=?openpyxl.load_workbook(url) ????#?根據指定表名獲取表格并得到對應的sheet對象 ????sheet?=?workbook[sheet_name] ????for?tdd?in?two_dimensional_data: ????????sheet.append(tdd) ????#?保存到指定位置 ????workbook.save(url) ????print("追加寫入成功")
3. 結果測試
3.1. 讀取測試
先準備兩個Excel文件,如圖所示
其內容如下:
測試代碼:
data1=read_xlsx_excel(r'D:\ii\aa.xlsx',"Sheet1") data2=read_xlsx_excel(r'D:\ii\bb.xls',1) for da in data1: print(da) print("------------") for da2 in data1: print(da2)
輸出結果:
結論:表明讀取并沒有問題!!!接下來測試寫入
3.2. 寫入測試
測試代碼:
uio=[[1,"adsf"],[5,3],[5,"你好"]] iuo=[[1,2],["adf",3],[5,"你好"]] wirte_xlsx_excel(r'D:\ii\aa.xlsx',"Sheet1",uio) wirte_xlsx_excel(r'D:\ii\bb.xls',"Sheet1",iuo)
結果輸出:
看到數據被覆蓋了?。?!接下來在上面修改后的數據的基礎上測試追加寫入
3.3. 追加寫入測試
測試代碼:
li=[[1,"add"],[1,"加入"]] write_xlsx_excel_add(r'D:\ii\bb.xls',li,1) kk=[[1,"woshilie"],[4,"列"]] write_xlsx_excel_add(r'D:\ii\aa.xlsx',"Sheet1",kk)
結果輸出:
追加也沒有問題?。?/p>
4. 總結
我們一直認為數據分析就應該以數據為主,而不應該把大量的代碼花費在設計Excel表格的樣式上,這樣多少就有點主客顛倒的意思了。
到此這篇關于Python操作Excel數據的封裝函數分享的文章就介紹到這了,更多相關Python操作Excel數據內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
一篇教程教你學會Python進制轉換(十進制轉二進制、八進制、十六進制)
計算機中只有1和0,所以就導致很多時候需要進制轉換,本文介紹了Python進制轉換,十進制轉二進制,十進制轉八進制,十進制轉十六進制,有興趣的可以了解一下2021-05-05PyCharm配置KBEngine快速處理代碼提示沖突、配置命令問題
這篇文章主要介紹了PyCharm配置KBEngine,解決代碼提示沖突、配置命令,本文通過圖文并茂的形式給大家介紹的超詳細,需要的朋友可以參考下2021-04-04使用Python通過win32 COM實現Word文檔的寫入與保存方法
今天小編就為大家分享一篇使用Python通過win32 COM實現Word文檔的寫入與保存方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05