openpyxl是一個第三方庫,可以處理xlsx格式的Excel檔案。pip install openpyxl安裝。
讀取Excel檔案
需要匯入相關函式
from openpyxl import load_workbook # 預設可讀寫,若有需要可以指定write_only和read_only為True wb = load_workbook('pythontab.xlsx') 預設開啟的檔案為可讀寫,若有需要可以指定引數read_only為True。
獲取工作表--Sheet
# 獲得所有sheet的名稱 print(wb.get_sheet_names()) # 根據sheet名字獲得sheet a_sheet = wb.get_sheet_by_name('Sheet1') # 獲得sheet名 print(a_sheet.title) # 獲得當前正在顯示的sheet, 也可以用wb.get_active_sheet() sheet = wb.active
獲取單元格
# 獲取某個單元格的值,觀察excel發現也是先字母再數字的順序,即先列再行 b4 = sheet['B4'] # 分別返回 print(f'({b4.column}, {b4.row}) is {b4.value}') # 返回的數字就是int型 # 除了用下標的方式獲得,還可以用cell函式, 換成數字,這個表示B2 b4_too = sheet.cell(row=4, column=2) print(b4_too.value)
b4.column返回B, b4.row返回4, value則是那個單元格的值。另外cell還有一個屬性coordinate, 像b4這個單元格返回的是座標B4。
獲得最大行和最大列
# 獲得最大列和最大行 print(sheet.max_row) print(sheet.max_column)
獲取行和列
sheet.rows為生成器, 裡面是每一行的資料,每一行又由一個tuple包裹。
sheet.columns類似,不過裡面是每個tuple是每一列的單元格。
# 因為按行,所以返回A1, B1, C1這樣的順序 for row in sheet.rows: for cell in row: print(cell.value) # A1, A2, A3這樣的順序 for column in sheet.columns: for cell in column: print(cell.value)
上面的程式碼就可以獲得所有單元格的資料。如果要獲得某行的資料呢?給其一個索引就行了,因為sheet.rows是生成器型別,不能使用索引,轉換成list之後再使用索引,list(sheet.rows)[2]這樣就獲取到第二行的tuple物件。
for cell in list(sheet.rows)[2]: print(cell.value)
如何獲得任意區間的單元格?
可以使用range函式,下面的寫法,獲得了以A1為左上角,B3為右下角矩形區域的所有單元格。注意range從1開始的,因為在openpyxl中為了和Excel中的表達方式一致,並不和程式語言的習慣以0表示第一個值。
for i in range(1, 4): for j in range(1, 3): print(sheet.cell(row=i, column=j)) # out <Cell pythontab.A1> <Cell pythontab.B1> <Cell pythontab.A2> <Cell pythontab.B2> <Cell pythontab.A3> <Cell pythontab.B3>
還可以像使用切片那樣使用。sheet['A1':'B3']返回一個tuple,該元組內部還是元組,由每行的單元格構成一個元組。
for row_cell in sheet['A1':'B3']: for cell in row_cell: print(cell) for cell in sheet['A1':'B3']: print(cell) # out (<Cell pythontab.A1>, <Cell pythontab.B1>) (<Cell pythontab.A2>, <Cell pythontab.B2>) (<Cell pythontab.A3>, <Cell pythontab.B3>)
根據字母獲得列號,根據列號返回字母
需要匯入, 這兩個函式存在於openpyxl.utils
from openpyxl.utils import get_column_letter, column_index_from_string # 根據列的數字返回字母 print(get_column_letter(2)) # B # 根據字母返回列的數字 print(column_index_from_string('D')) # 4
將資料寫入Excel
工作表相關
需要匯入WorkBook
from openpyxl import Workbook wb = Workbook()
這樣就新建了一個新的工作表(只是還沒被儲存)。
若要指定只寫模式,可以指定引數write_only=True。一般預設的可寫可讀模式就可以了。
print(wb.get_sheet_names()) # 提供一個預設名叫Sheet的表,office2016下新建提供預設Sheet1 # 直接賦值就可以改工作表的名稱 sheet.title = 'Sheet1' # 新建一個工作表,可以指定索引,適當安排其在工作簿中的位置 wb.create_sheet('Data', index=1) # 被安排到第二個工作表,index=0就是第一個位置 # 刪除某個工作表 wb.remove(sheet) del wb[sheet]
寫入單元格
還可以使用公式
# 直接給單元格賦值就行 sheet['A1'] = 'good' # B9處寫入平均值 sheet['B9'] = '=AVERAGE(B2:B8)'
但是如果是讀取的時候需要加上data_only=True這樣讀到B9返回的就是數字,如果不加這個引數,返回的將是公式本身'=AVERAGE(B2:B8)'
append函式
可以一次新增多行資料,從第一行空白行開始(下面都是空白行)寫入。
# 新增一行 row = [1 ,2, 3, 4, 5] sheet.append(row) # 新增多行 rows = [ ['Number', 'data1', 'data2'], [2, 40, 30], [3, 40, 25], [4, 50, 30], [5, 30, 10], [6, 25, 5], [7, 50, 10], ]
由於append函式只能按行寫入。如果我們想按列寫入呢。append能實現需求麼?如果把上面的列表巢狀看作矩陣。只要將矩陣轉置就可以了。使用zip()函式可以實現,不過內部的列表變成了元組就是了。都是可迭代物件,不影響。
list(zip(*rows)) # out [('Number', 2, 3, 4, 5, 6, 7), ('data1', 40, 40, 50, 30, 25, 50), ('data2', 30, 25, 30, 10, 5, 10)]
解釋下上面的list(zip(*rows))首先*rows將列表打散,相當於填入了若干個引數,zip從某個列表中提取第1個值組合成一個tuple,再從每個列表中提取第2個值組合成一個tuple,一直到最短列表的最後一個值提取完畢後結束,更長列表的之後的值被捨棄,換句話,最後的元組個數是由原來每個引數(可迭代物件)的最短長度決定的。比如現在隨便刪掉一個值,最短列表長度為2,data2那一列(豎著看)的值全部被捨棄。
rows = [ ['Number', 'data1', 'data2'], [2, 40], [3, 40, 25], [4, 50, 30], [5, 30, 10], [6, 25, 5], [7, 50, 10], ] # out [('Number', 2, 3, 4, 5, 6, 7), ('data1', 40, 40, 50, 30, 25, 50)]
最後zip返回的是zip物件,看不到資料的。使用list轉換下就好了。使用zip可以方便實現將資料按列寫入。
儲存檔案
所有的操作結束後,一定記得儲存檔案。指定路徑和檔名,字尾名為xlsx。
wb.save(r'D:\example.xlsx')
設定單元格風格--Style
先匯入需要的類
from openpyxl.styles import Font, colors, Alignment
分別可指定字型相關,顏色,和對齊方式。
字型
bold_itatic_24_font = Font(name='等線', size=24, italic=True, color=colors.RED, bold=True) sheet['A1'].font = bold_itatic_24_font
上面的程式碼指定了等線24號加粗斜體,字型顏色紅色。直接使用cell的font屬性,將Font物件賦值給它。
對齊方式
也是直接使用cell的屬性aligment,這裡指定垂直居中和水平居中。除了center,還可以使用right、left等等引數。
# 設定B1中的資料垂直居中和水平居中 sheet['B1'].alignment = Alignment(horizontal='center', vertical='center')
設定行高和列寬
有時候資料太長顯示不完,就需要拉長拉高單元格。
# 第2行行高 sheet.row_dimensions[2].height = 40 # C列列寬 sheet.column_dimensions['C'].width = 30
合併和拆分單元格
所謂合併單元格,即以合併區域的左上角的那個單元格為基準,覆蓋其他單元格使之稱為一個大的單元格。
相反,拆分單元格後將這個大單元格的值返回到原來的左上角位置。
# 合併單元格, 往左上角寫入資料即可 sheet.merge_cells('B1:G1') # 合併一行中的幾個單元格 sheet.merge_cells('A1:C3') # 合併一個矩形區域中的單元格
合併後只可以往左上角寫入資料,也就是區間中:左邊的座標。
如果這些要合併的單元格都有資料,只會保留左上角的資料,其他則丟棄。換句話說若合併前不是在左上角寫入資料,合併後單元格中不會有資料。
以下是拆分單元格的程式碼。拆分後,值回到A1位置。
sheet.unmerge_cells('A1:C3')