python日常實用技能:使用python將大量數據導出到Excel中的

@本文來源於公衆號:csdn2299,喜歡可以關注公衆號 程序員學府
(1) 問題描述:爲了更好地展示數據,Excel格式的數據文件往往比文本文件更具有優勢,但是具體到python中,該如何導出數據到Excel呢?如果碰到需要導出大量數據又該如何操作呢?

本文主要解決以上兩個問題。

(2)具體步驟如下:

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

2.第二步,哈哈,沒有啦,廢話不說了,直接上代碼,ps,代碼中包含xlwt和openpyxl的兩個實現版本。

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

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的情況,你將有如下三種選擇,

(1) 換一種存儲格式,如保存爲CSV文件

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

(3) win32 COM (Windows only)

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

ps,非常lucky,一番搜索後我找到了openpyxl,支持07+的excel,一直有人在維護,文檔清晰易讀,參照Tutorial和API文檔很快就能上手,就是它了~

(4) 閒話少說,直接上代碼,敬請參考

# coding:utf-8
'''
# 希望對大家有幫助哈,請多提問題
create by yaoyz
date: 2017/01/24
'''
import xlrd
import xlwt
# workbook相關
from openpyxl.workbook import Workbook
# ExcelWriter,封裝了很強大的excel寫的功能
from openpyxl.writer.excel import ExcelWriter
# 一個eggache的數字轉爲列字母的方法
from openpyxl.utils import get_column_letter
from openpyxl.reader.excel import load_workbook
class HandleExcel():
 '''Excel相關操作類'''
 def __init__(self):
  self. head_row_labels = [u'學生ID',u'學生姓名',u'聯繫方式',u'知識點ID',u'知識點名稱']
 """
  function:
   讀出txt文件中的每一條記錄,把它保存在list中
  Param:
   filename: 要讀出的文件名
  Return:
   res_list: 返回的記錄的list
 """
 def read_from_file(self,filename):
  res_list=[]
  file_obj=open(filename,"r")
  for line in file_obj.readlines():
   res_list.append(line)
  file_obj.close()
  return res_list
 """
  function:
   讀出*.xlsx中的每一條記錄,把它保存在data_dic中返回
  Param:
   excel_name: 要讀出的文件名
  Return:
   data_dic: 返回的記錄的dict
 """
 def read_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
  # 把數據存到字典中
  for row in ws.rows:
   temp_list=[]
   # print "row",row
   for cell in row:
     print cell.value,
     temp_list.append(cell.value)
   print ""
   data_dic[sign]=temp_list
   sign+=1
  print data_dic
  return data_dic
 """
  function:
   讀出*.xlsx中的每一條記錄,把它保存在data_dic中返回
  Param:
   records: 要保存的,一個包含每一條記錄的list
   save_excel_name: 保存爲的文件名
   head_row_stu_arrive_star:
  Return:
   data_dic: 返回的記錄的dict
 """
 def write_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"
  # 寫第一行,標題行
  for h_x in range(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
  for record in records:
   record_list=str(record).strip().split("\t")
   for x in range(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:
   無
 """
 def read_excel(self,excel_name):
  workbook=xlrd.open_workbook(excel_name)
  print workbook.sheet_names()
  # 獲取所有sheet
  print workbook.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的名稱,行數,列數
  print sheet2.name,sheet2.nrows,sheet2.ncols
  # 獲取整行和整列的值(數組)
  rows = sheet2.row_values(3) # 獲取第四行內容
  cols = sheet2.col_values(2) # 獲取第三列內容
  print rows
  print cols
  # 獲取單元格內容
  print sheet2.cell(1,0).value
  print sheet2.cell_value(1,0)
  print sheet2.row(1)[0].value
  # 獲取單元格內容的數據類型
  print sheet2.cell(1,0).ctype
  # 通過名稱獲取
  return workbook.sheet_by_name(u'Sheet1')
 """
  function:
   設置單元格樣式
  Param:
   name: 字體名字
   height: 字體高度
   bold: 是否大寫
  Return:
   style: 返回設置好的格式對象
 """
 def set_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
  return style
 """
  function:
   按照 設置單元格樣式 把計算結果由txt轉變爲Excel存儲
  Param:
   dataset:要保存的結果數據,list存儲
  Return:
   將結果保存爲 excel對象中
 """
 def write_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
  # 首行標題:
  for p in range(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
  for line in dataset:
   row_list=str(line).strip("\n").split("\t")
   for pp in range(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) # 保存文件
 def run_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文件的任務"
 def run_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,工作了發現吃了計算機基礎不好的虧,學歷不行這是

沒辦法的事,只能後天彌補,於是在編碼之外開啓了自己的逆襲之路,不斷的學習python核心知識,深

入的研習計算機基礎知識,整理好了,我放在我們的微信公衆號《程序員學府》,如果你也不甘平庸,之外的東西,比如,如何做一個精緻的程序員,而不是“屌絲”,程序員本身就是高貴的一種存在啊,難道不是嗎?[點擊加入]想做你自己想成爲高尚人,加油!

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