Python 利用xlwt模塊實(shí)現(xiàn)將數(shù)據(jù)寫入excel文件

Python 利用xlwt模塊實(shí)現(xiàn)將數(shù)據(jù)寫入excel文件？很多新手對(duì)此不是很清楚，為了幫助大家解決這個(gè)難題，下面小編將為大家詳細(xì)講解，有這方面需求的人可以來學(xué)習(xí)下，希望你能有所收獲。

#!/usr/bin/python
# -*- coding: utf-8 -*-

import xlwt
import re

def host_regex(dataline):
 host_regex = r"<host>(.*&#63;)</host>"
 host = re.findall(host_regex, dataline)
 if host:
  return host[0]

def ip_regex(dataline):
 ip_regex = r"<ip>(.*&#63;)</ip>"
 ip = re.findall(ip_regex, dataline)
 if ip:
  return ip[0]

with open("zbx_export_hosts.xml", "r", encoding="utf8") as f:
 lines = f.readlines()
 alldatas = []
 hostlist = []
 iplist = []
 for line in lines:
  host = host_regex(line)
  ip = ip_regex(line)

  if host is not None:
   hostlist.append(host)
  if ip is not None:
   iplist.append(ip)
 hosts_ip = []
 # 構(gòu)造數(shù)據(jù)結(jié)構(gòu),形如：[[a,b],[c,d],...]
 multi_list = map(list, zip(hostlist, iplist))
 for multi in multi_list:
  hosts_ip.append(multi)
 workbook = xlwt.Workbook()
 worksheet = workbook.add_sheet('test')
 # 關(guān)鍵代碼
 for hi in enumerate(hosts_ip):
  for num in range(len(hi[1])):
   # 行、列、值
   worksheet.write(hi[0], num, hi[1][num])

 workbook.save('excelwrite.xls')

PS：其實(shí)有兩行代碼冗余，就不刪了！

補(bǔ)充知識(shí)：通過python寫入xlsx大量數(shù)據(jù)問題簡(jiǎn)述

以前批處理生成數(shù)據(jù)文件都是通過c程序直接按照逗號(hào)分隔方式寫文本文件以csv保存的。但是由于有些處理的數(shù)據(jù)表直接給業(yè)務(wù)人員看，業(yè)務(wù)人員習(xí)慣使用xls，因此提了些需求改造，對(duì)于一些小于100w的數(shù)據(jù)均要求改為xlsx文件格式的報(bào)表。

考慮報(bào)表文件數(shù)據(jù)處理便捷，因此決定用python實(shí)現(xiàn)，最初使用了openpyxl庫，這個(gè)python庫擁有非常強(qiáng)大的xlsx的讀寫api，能夠很方便的實(shí)現(xiàn)讀寫。但是在實(shí)際使用中發(fā)現(xiàn)在處理較大行數(shù)的xlsx時(shí)，占用內(nèi)存非常嚴(yán)重，8個(gè)字段50w行數(shù)據(jù)在運(yùn)行中內(nèi)存高達(dá)10G，這不是我們應(yīng)用服務(wù)器能夠接受的范圍。

因此尋求改善，在網(wǎng)上了解后，發(fā)現(xiàn)創(chuàng)建xlsx表時(shí)，指定write_only參數(shù)是openpyxl提供的大數(shù)據(jù)寫入解決方案，以下摘自官網(wǎng)范例，在選擇了此參數(shù)后，無法通過ws.cell(row=i, column=j).value = ？ 的方式復(fù)制，只能通過append的方式追加行。關(guān)于設(shè)置只寫cell設(shè)置樣式，可以直接查看官網(wǎng)

>>> from openpyxl import Workbook
>>> wb = Workbook(write_only=True)
>>> ws = wb.create_sheet()
>>>
>>> # now we'll fill it with 100 rows x 200 columns
>>>
>>> for irow in range(100):
...  ws.append(['%d' % i for i in range(200)]) #可以先將一行的數(shù)據(jù)寫在一個(gè)list中，直接ws.append(list)
>>> # save the file
>>> wb.save('new_big_file.xlsx') # doctest: +SKIP

不過僅僅按上述范例修改write_only跑數(shù)時(shí)發(fā)現(xiàn)內(nèi)存占用并沒有緩解，再仔細(xì)閱讀官網(wǎng)說明時(shí)才發(fā)現(xiàn)這么一句話“When you want to dump large amounts of data make sure you have lxml installed.”原來在openpyxl安裝時(shí)，并沒有把lxml作為強(qiáng)依賴，只是一個(gè)可選包，然后查實(shí)發(fā)現(xiàn)確實(shí)服務(wù)器上并沒有安裝lxml包。

于是與為了解決這個(gè)問題先在測(cè)試環(huán)境上安裝lxml，安裝過程也不是很順利。

首先先要裝libxml2以及l(fā)ibxslt，因?yàn)闆]有root權(quán)限，均為源碼安裝，且只能安裝在用戶目錄。需要注意的是libxml2在安裝時(shí)需要--with-python='python安裝目錄'，libxslt安裝的時(shí)候需要指定libxml2安裝目錄。

尤其最后的lxml安裝也是坑挺多的，主要是指定libxml2和libxslt，安裝lxml時(shí)先python3 setup.py build_ext -i -I${libxml2includepath} -L${libxml2libpath} -I${libxsltincludepath} -L${libxsltlibpath} ，再python3 setup.py install。

全部安裝完之后，確實(shí)openpyxl寫xlsx不再增加內(nèi)存，先在tmp目錄一個(gè)臨時(shí)文件中寫，待調(diào)用save方法時(shí)，保存到正式文件。

此外在這個(gè)問題解決中，還了解到writexlsx庫也可以處理大文件xlsx的寫入，并且安裝沒有這么多的依賴，其實(shí)還是非常推薦，也很簡(jiǎn)介，官網(wǎng)有介紹。

操作大文件時(shí)指定{'constant_memory': True}即可。

filename = 'test.xlsx'
wb = xlsxwriter.Workbook(filename, {'constant_memory': True})
ws = wb.add_worksheet('大文件')
i = 0
data= []
data.append('中文')
data.append(1323)
data.append('abc')
ws.write_row(i, 0, data)
i = i + 1
wb.close() #直接關(guān)閉即可，不需要額外save

看完上述內(nèi)容是否對(duì)您有幫助呢？如果還想對(duì)相關(guān)知識(shí)有進(jìn)一步的了解或閱讀更多相關(guān)文章，請(qǐng)關(guān)注億速云行業(yè)資訊頻道，感謝您對(duì)億速云的支持。