Python 批量插入100萬級數據到Excel文件(簡潔版)


  引言

  python是一門開發語言,可以用來寫大型項目,也可以用來寫腳本,比如自動化腳本,也可以用來寫工具。

  背景

  我們平時做測試,不一定做自動化測試才用去python,做功能測試,也就是點點點,也可以把它用起來。

  比如,我現在頁面有一個功能,導入的功能,我想測它最大支持多少數據量導入,或者說多大的文件導入,那么我們得在excel中寫入很多條數據(導入模板是xls,xlsx)。

  如果是自己手一條一條去弄,顯然不現實,數據量不多還好,如果是100萬呢,500萬呢,1000萬呢....

  那么,這個時候可以利用python寫個腳本,一秒生成百萬級+的excel文件。

  案例實現

  我就將我實際項目遇到的情況拿出來講。

  背景:

  1、導入功能;

  2、導入模板excel;

  3、導入數據格式,兩列數據,數據格式不限制。

  根據需求,我想測這個導入支持多大的數據量以及達到最大的時候,頁面提示什么信息,是報錯,還是溫馨提示呢?那么就得寫這樣的一個向excel批量插入數據的腳本:

from xlwt import Workbook

#創建一個工作簿
w = Workbook()
#創建一個工作表
ws = w.add_sheet('1')
# 計數
count = 0
# 行數
link_nums = 5000
for j in range(0,2):     #控制列
    for i in range(0, link_nums):   #控制行
        count += 1

        if(j == 0):         #第一列
            ws.write(i, j, i)
        if(j == 1):
            ws.write(i,j,i)


print("總共生成(單元格): %s數據"%count)
print("總共生成: %s條數據"%link_nums)

w.save('xqtest.xls')

  

  運行結果:

總共生成(單元格): 10000數據
總共生成: 5000條數據

Process finished with exit code 0

  打開excel文件查看:

 

 運行時長:

總共生成(單元格): 10000數據
總共生成: 5000條數據
Running time: 7.293443303871468e-07 Seconds 

Process finished with exit code 0

 

  拿着生成的excel文件,操作導入,結果報錯:

 

  然后測試了這個導入功能,果然是有bug,已經反饋給開發修改了。

 

  總結

 

  方法基本上就是這樣,具體根據自己業務需要進行設計。

  另外,對測試開發,自動化測試,全棧測試相關技術感興趣的朋友,可以加入到群里學習和探索交流,進群方式,掃下方二維碼。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM