【好消息!】Python和Excel終於可以互通了!!

今天小編爲大家分享一篇使用python將大量數據導出到Excel中的技巧心得,可以讓Python和Excel的數據實現互通!具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧!

問題描述

爲了更好地展示數據,Excel格式的數據文件往往比文本文件更具有優勢,但是具體python中,該如何導出數據到Excel呢?如果碰到需要導出大量數據又該如何操作呢?

具體步驟

Step 1 安裝openpyxl

使用pip install openpyxl即可,但是在windows下安裝的是2.2.6版本,在centos自動安裝的是4.1版本,寫的代碼在windows下運行沒問題,但centos上卻報錯了,說是ew=ExcelWriter(workbook=wb)少提供一個參數,於是果斷在 237服務器上我已安裝2.2.6版本的,問題解決。

pip install openpyxl==2.2.6

Step 2 直接上代碼(Ps:代碼中包含xlwt和openpyxl的兩個實現版本)

# coding:utf-8
'''
# 希望對大家有幫助哈,請多提問題
create by yaoyz
date: 2017/01/24
'''
importxlrd
importxlwt
# workbook相關
fromopenpyxl.workbookimportWorkbook
# ExcelWriter,封裝了很強大的excel寫的功能
fromopenpyxl.writer.excelimportExcelWriter
# 一個eggache的數字轉爲列字母的方法
fromopenpyxl.utilsimportget_column_letter
fromopenpyxl.reader.excelimportload_workbook
classHandleExcel():
'''Excel相關操作類'''
def__init__(self):
self. head_row_labels = [u'學生ID',u'學生姓名',u'聯繫方式',u'知識點ID',u'知識點名稱']
"""
function:
讀出txt文件中的每一條記錄,把它保存在list中
Param:
filename:  要讀出的文件名
Return:
res_list:返回的記錄的list
"""
defread_from_file(self,filename):
res_list=[]
file_obj=open(filename,"r")
forlineinfile_obj.readlines():
res_list.append(line)
file_obj.close()
returnres_list
"""
function:
讀出*.xlsx中的每一條記錄,把它保存在data_dic中返回
Param:
excel_name:  要讀出的文件名
Return:
data_dic:返回的記錄的dict
"""
defread_excel_with_openpyxl(self, excel_name="testexcel2007.xlsx"):
# 讀取excel2007文件
wb = load_workbook(filename=excel_name)
# 顯示有多少張表
print"Worksheet range(s):", wb.get_named_ranges()
print"Worksheet name(s):", wb.get_sheet_names()
# 取第一張表
sheetnames = wb.get_sheet_names()
ws = wb.get_sheet_by_name(sheetnames[0])
# 顯示錶名,錶行數,表列數
print"Work Sheet Titile:",ws.title
print"Work Sheet Rows:",ws.get_highest_row()
print"Work Sheet Cols:",ws.get_highest_column()
# 獲取讀入的excel表格的有多少行,有多少列
row_num=ws.get_highest_row()
col_num=ws.get_highest_column()
print"row_num: ",row_num," col_num: ",col_num
# 建立存儲數據的字典
data_dic = {}
sign=1
# 把數據存到字典中
forrowinws.rows:
temp_list=[]
# print "row",row
forcellinrow:
printcell.value,
temp_list.append(cell.value)
print""
data_dic[sign]=temp_list
sign+=1
printdata_dic
returndata_dic
"""
function:
讀出*.xlsx中的每一條記錄,把它保存在data_dic中返回
Param:
records: 要保存的,一個包含每一條記錄的list
save_excel_name:  保存爲的文件名
head_row_stu_arrive_star:
Return:
data_dic:返回的記錄的dict
"""
defwrite_to_excel_with_openpyxl(self,records,head_row,save_excel_name="save.xlsx"):
# 新建一個workbook
wb = Workbook()
# 新建一個excelWriter
ew = ExcelWriter(workbook=wb)
# 設置文件輸出路徑與名稱
dest_filename = save_excel_name.decode('utf-8')
# 第一個sheet是ws
ws = wb.worksheets[0]
# 設置ws的名稱
ws.title ="range names"
# 寫第一行,標題行
forh_xinrange(1,len(head_row)+1):
h_col=get_column_letter(h_x)
#print h_col
ws.cell('%s%s'% (h_col,1)).value ='%s'% (head_row[h_x-1])
# 寫第二行及其以後的那些行
i =2
forrecordinrecords:
record_list=str(record).strip().split("\t")
forxinrange(1,len(record_list)+1):
col = get_column_letter(x)
ws.cell('%s%s'% (col, i)).value ='%s'% (record_list[x-1].decode('utf-8'))
i +=1
# 寫文件
ew.save(filename=dest_filename)
"""
function:
測試輸出Excel內容
讀出Excel文件
Param:
excel_name:  要讀出的Excel文件名
Return:
無
"""
defread_excel(self,excel_name):
workbook=xlrd.open_workbook(excel_name)
printworkbook.sheet_names()
# 獲取所有sheet
printworkbook.sheet_names()# [u'sheet1', u'sheet2']
sheet2_name = workbook.sheet_names()[1]
# 根據sheet索引或者名稱獲取sheet內容
sheet2 = workbook.sheet_by_index(1)# sheet索引從0開始
sheet2 = workbook.sheet_by_name('Sheet1')
# sheet的名稱,行數,列數
printsheet2.name,sheet2.nrows,sheet2.ncols
# 獲取整行和整列的值(數組)
rows = sheet2.row_values(3)# 獲取第四行內容
cols = sheet2.col_values(2)# 獲取第三列內容
printrows
printcols
# 獲取單元格內容
printsheet2.cell(1,0).value
printsheet2.cell_value(1,0)
printsheet2.row(1)[0].value
# 獲取單元格內容的數據類型
printsheet2.cell(1,0).ctype
# 通過名稱獲取
returnworkbook.sheet_by_name(u'Sheet1')
"""
function:
設置單元格樣式
Param:
name:  字體名字
height:  字體高度
bold:  是否大寫
Return:
style: 返回設置好的格式對象
"""
defset_style(self,name,height,bold=False):
style = xlwt.XFStyle()# 初始化樣式
font = xlwt.Font()# 爲樣式創建字體
font.name = name# 'Times New Roman'
font.bold = bold
font.color_index =4
font.height = height
borders= xlwt.Borders()
borders.left=6
borders.right=6
borders.top=6
borders.bottom=6
style.font = font
style.borders = borders
returnstyle
"""
function:
按照 設置單元格樣式  把計算結果由txt轉變爲Excel存儲
Param:
dataset:要保存的結果數據,list存儲
Return:
將結果保存爲 excel對象中
"""
defwrite_to_excel(self, dataset,save_excel_name,head_row):
f = xlwt.Workbook()# 創建工作簿
# 創建第一個sheet:
# sheet1
count=1
sheet1 = f.add_sheet(u'sheet1', cell_overwrite_ok=True)# 創建sheet
# 首行標題:
forpinrange(len(head_row)):
sheet1.write(0,p,head_row[p],self.set_style('Times New Roman',250,True))
default=self.set_style('Times New Roman',200,False)# define style out the loop will work
forlineindataset:
row_list=str(line).strip("\n").split("\t")
forppinrange(len(str(line).strip("\n").split("\t"))):
sheet1.write(count,pp,row_list[pp].decode('utf-8'),default)
count+=1
f.save(save_excel_name)# 保存文件
defrun_main_save_to_excel_with_openpyxl(self):
print"測試讀寫2007及以後的excel文件xlsx,以方便寫入文件更多數據"
print"1. 把txt文件讀入到內存中,以list對象存儲"
dataset_list=self.read_from_file("test_excel.txt")
'''test use openpyxl to handle EXCEL 2007'''
print"2. 把文件寫入到Excel表格中"
head_row_label=self.head_row_labels
save_name="test_openpyxl.xlsx"
self.write_to_excel_with_openpyxl(dataset_list,head_row_label,save_name)
print"3.  執行完畢,由txt格式文件保存爲Excel文件的任務"
defrun_main_save_to_excel_with_xlwt(self):
print" 4. 把txt文件讀入到內存中,以list對象存儲"
dataset_list=self.read_from_file("test_excel.txt")
'''test use xlwt to handle EXCEL 97-2003'''
print" 5. 把文件寫入到Excel表格中"
head_row_label=self.head_row_labels
save_name="test_xlwt.xls"
self.write_to_excel_with_openpyxl(dataset_list,head_row_label,save_name)
print"6.  執行完畢,由txt格式文件保存爲Excel文件的任務"
if__name__ =='__main__':
print"create handle Excel Object"
obj_handle_excel=HandleExcel()
# 分別使用openpyxl和xlwt將數據寫入文件
obj_handle_excel.run_main_save_to_excel_with_openpyxl()
obj_handle_excel.run_main_save_to_excel_with_xlwt()
'''測試讀出文件,注意openpyxl不可以讀取xls的文件,xlrd不可以讀取xlsx格式的文件'''
#obj_handle_excel.read_excel_with_openpyxl("testexcel2003.xls")  # 錯誤寫法
#obj_handle_excel.read_excel_with_openpyxl("testexcel2003.xls") # 錯誤寫法
obj_handle_excel.read_excel("testexcel2003.xls")
obj_handle_excel.read_excel_with_openpyxl("testexcel2007.xlsx")

 擴展閱讀

通過查閱資料,發現網上衆說紛紜,總結起來有如下幾點:

python Excel相關操作的module lib有兩組,一組是xlrd、xlwt、xlutils,另一組是openpyxl,但是前一組(xlrd,xlwt)比較老,只能處理由Excel 97-2003 或者Excel 97 以前版本生成的xls格式的excel文件,xlwt甚至不支持07版以後的excel,這個格式excel文件一般來說,最大隻能支持256列或者65536行的excel文件。

因此面對需要導出大量數據到excel的情況,你將有如下三種選擇:

  • 換一種存儲格式,如保存爲CSV文件

  • 使用openpyxl—,因爲它支持對Excel 2007+ xlsx/xlsm format的處理

  • win32 COM (Windows only)

當然,我們要直面困難,爲了更好地展示數據給產品和用戶,我們依然選擇第二種。

經過一番搜索後我找到了openpyxl的網址,放在下面了,支持07+的excel,一直有人在維護,文檔清晰易讀,參照Tutorial和API文檔很快就能上手了,大家有需要的可以自取。

  • openpyxl網址:

https://openpyxl.readthedocs.io/en/stable/(可以將網頁翻譯成中文方便使用)

End.

作者:小天真_5eed

來源:簡書

本文爲轉載分享,如有侵權請聯繫後臺刪除

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章