點擊上方「AirPython」,選擇「加爲星標」web
第一時間關注 Python 技術乾貨!sql
1. 前言
在咱們平常工做中,常常會使用 Word、Excel、PPT、PDF 等辦公軟件微信
可是,常常會遇到一些重複繁瑣的事情,這時候手工操做顯得效率極其低下;經過 Python 實現辦公自動化變的頗有必要app
接下來的 一系列 文章,我將帶你們對 Python 辦公自動化作一個全面的總結,絕對的乾貨!函數
2. 準備
使用 Python 操做 Excel 文件,常見的方式以下:字體
xlrd / xlwt ui
openpyxlspa
Pandas.net
xlsxwriter3d
xlwings
pywin32
xlrd 和 xlwt 是操做 Excel 文件最多的兩個依賴庫
其中,
xlrd 負責讀取 Excel 文件,xlwt 能夠寫入數據到 Excel 文件
咱們安裝這兩個依賴庫
# 安裝依賴庫
pip3 install xlrd
pip3 install xlwt
3. xlrd 讀取 Excel
使用 xlrd 中的 open_workbook(filepath) 打開本地一個 Excel 文件
import xlrd
# 打開文件,返回一個工做簿對象
wb = xlrd.open_workbook(file_path)
工做簿對象的 nsheets 屬性獲取 Sheet 數目,sheet_names() 方法返回全部 Sheet 名稱的列表
# 統計sheet數量
sheets_num, sheets_names = wb.nsheets, wb.sheet_names()
print('sheet數量一共有:', sheets_num)
print('sheet名稱分別爲:', sheets_names)
篩選出工做簿中的某一個 Sheet 有 2 種方式,分別是:
經過 Sheet 名稱
使用位置索引,從 0 開始
# 獲取某一個sheet
# 經過名稱或者索引獲取
sheet = wb.sheet_by_index(0)
# sheet = wb.sheet_by_name('第一個Sheet')
print(sheet)
每個 sheet 對象均可以利用 name、nrows、ncols 獲取 Sheet 名稱、行數量、列數量
另外
row_values(index)、col_values(index) 分別用於獲取某一行或某一列的數據列表
# 獲取某一個sheet中,包含的行數量、列數量
sheet_name, sheet_row_count, sheet_column_count = sheet.name, sheet.nrows, sheet.ncols
print('當前sheet名稱爲:', sheet_name, ",一共有:", sheet_row_count, "行;有:", sheet_column_count, "列")
# 單獨獲取某一行數據,索引從0開始
# 好比:獲取第2行數據
row_datas = sheet.row_values(1)
print('第2行數據爲:', row_datas)
# 單獨獲取某一列數據,索引從0開始
# 好比:獲取第二列數據
column_datas = sheet.col_values(1)
print('第2列數據爲:', column_datas)
單元格能夠經過行索引、列索引,調用 cell(row_index,column_index) 函數獲取
須要注意的是,行索引和列索引都是從 0 開始,即:0 表明第一行
在 xlrd 中,單元格的數據類型包含 6 種,用 ctype 屬性對應關係以下:
0 -- 空(empty)
1 -- 字符串(string)
2 -- 數字(number)
3 -- date(日期)
4 -- boolean(布爾值)
5 -- error(錯誤)
# 獲取某一個單元格的數據
# 好比:獲取第2行第1列的單元格的數據
one_cell = sheet.cell(1, 0)
# 單元格的值
cell_value = one_cell.value
print("單元格的值爲:", cell_value)
# 單元格數據類型
cell_type = one_cell.
print("單元格數據類型爲:", cell_type)
最後,若是要獲取當前 Sheet 全部單元格中的數據,能夠經過遍歷全部行、列來操做
# 獲取全部單元格的值
print('表格中全部數據以下:')
for r in range(sheet.nrows):
for i in range(sheet.ncols):
print(sheet.cell(r, i).value)
4. xlwt 寫入 Excel
若是想實現將數據寫入到 Excel 中,xlwt 就很方便了
首先,使用 xlwt 的 Workbook() 方法建立一個工做簿對象
而後,使用工做簿對象的 add_sheet(sheetname) 方法新增 Sheet
import xlwt
sheetname = '第一個Sheet'
# 建立一個工做簿對象
wb = xlwt.Workbook()
# 添加Sheet,經過sheet名稱
sheet = wb.add_sheet(sheetname)
接着,經過 sheet 對象的 write() 方法,按照行索引和列索引,將數據寫入到對應單元格中去
# 將數據寫入到Sheet中
# 3個參數分別是:行索引(從0開始)、列索引(從0開始)、單元格的值
# 第一行第一列,寫入一個數據
# 寫入標題
for index, title in enumerate(self.titles):
sheet.write(0, index, title)
# 寫入值
for index_row, row_values in enumerate(self.values):
for index_column, column_value in enumerate(row_values):
sheet.write(index_row + 1, index_column, column_value)
須要注意的是,最後必須調用工做簿的 save(filepath),才能在本地生成 Excel 文件
# 保存文件
# 最後保存文件便可
wb.save(filepath)
5. 進階用法
接下來,聊聊幾個經常使用的進階用法
一、獲取全部可見的 Sheet
在讀取 Sheet 數據時,常常須要過濾隱藏的 Sheet
當 sheet 對象的 visibility 屬性值爲 0 時,表明此 Sheet 在工做簿中是顯示的;不然被隱藏了
def get_all_visiable_sheets(self, wb):
"""
獲取全部可見的sheet
:param wb:
:return:
"""
return list(filter(lambda item: item.visibility == 0, wb.sheets()))
# 一、獲取全部可看見的sheet
sheet_visiable = self.get_all_visiable_sheets(wb)
print('全部可見的sheet包含:', sheet_visiable)
二、獲取 Sheet 可見行或列
某一個 Sheet 中,可能存在部分行、列被隱藏了
def get_all_visiable_rows(self, sheet):
"""
獲取某一個sheet中,可見的行
:param sheet:
:return:
"""
result = [index for index in range(sheet.nrows) if sheet.rowinfo_map[index].hidden == 0]
return result
def get_all_visiable_columns(self, sheet):
"""
獲取某一個sheet中,可見的列
:param sheet:
:return:
"""
result = [index for index in range(sheet.ncols) if sheet.colinfo_map[index].hidden == 0]
return result
三、獲取單元格的樣式
以獲取單元格字體顏色和背景爲例
def get_cell_bg_color(self, wb, sheet, row_index, col_index):
"""
獲取某一個單元格的背景顏色
:param wb:
:param sheet:
:param row_index:
:param col_index:
:return:
"""
xfx = sheet.cell_xf_index(row_index, col_index)
xf = wb.xf_list[xfx]
# 字體顏色
font_color = wb.font_list[xf.font_index].colour_index
# 背景顏色
bg_color = xf.background.pattern_colour_index
return font_color, bg_color
須要注意的是,使用 xlrd 讀取單元格的樣式,打開工做簿的時候須要顯式定義 formatting_info = True,不然會拋出異常
# 注意:必須設置formatting_info=True,才能正常獲取屬性
wb = xlrd.open_workbook(file_path, formatting_info=True)
sheet = wb.sheet_by_index(0
6. 最後
搭配使用 xlrd、xlwt,基本上能完成大部分的工做,對於一些複雜的功能,好比:複製、分割、篩選等功能,能夠用上 xlutils 這個依賴庫
須要指出的是,這個組合對 xlsx 的兼容性不太好;若是須要操做 xlsx 文件,須要先轉爲 xls,而後再進行
我已經將文中所有源碼上傳到後臺,關注公衆號後回覆「 excel 」便可得到所有源碼
若是你以爲文章還不錯,請你們 點贊、分享、留言 下,由於這將是我持續輸出更多優質文章的最強動力!
本文分享自微信公衆號 - 看那個碼農(gh_31e9c1ac7ce4)。
若有侵權,請聯繫 support@oschina.cn 刪除。
本文參與「OSC源創計劃」,歡迎正在閱讀的你也加入,一塊兒分享。