首頁 > 軟體

Python 如何實時向檔案寫入資料(附程式碼)

2022-07-11 18:02:26

之前在做資料分析的過程中,需要對資料進行實時的寫入,比如對新生成的資料寫入之前已經生成的txt或csv檔案中。現在想想其實很簡單,所以做一個總結。

1:實時向csv檔案寫入資料

假設需要生成一張csv表,裡面的欄位對應一些資料,由於後續的過程中,不止一次寫入資料,那麼安全的做法是:

  • 首先寫入欄位;
  • 然後寫入資料(否則欄位也會每次被寫入)

步驟1:建立檔案並寫入欄位

import csv
with open("test3.csv","a",newline='') as csvfile: 
    writer = csv.writer(csvfile, delimiter=' ')
    writer.writerow(["index","a_name","b_name"])

上述程式碼引數解釋:

  • ​ test3.csv表示要建立一個test3.csv的檔案,注意:如果當前目錄下沒有這個檔案,則會自動生成test3.csv檔案,如果當前目錄下已經有了test3.csv的檔案,那麼在新建結束後,會將原始的test3.csv檔案覆蓋。
  • ​ "a"表示匯入的資料不會講test3.csv檔案中的原始資料覆蓋,即:在後面繼續新增,如果需要覆蓋,則將"a"改成"w"即可。
  • newline=’ ’表示不會以空行作為每一行的分割線,注意:這一行程式碼必須新增上,否則csv檔案中的每一行資料的前面會出現空行。

步驟2:寫入資料

list1=[0,0,0]
list2=[1,1,1]
data_array=[[5,5,5],[1,2,3]]
with open("test3.csv","a",newline='') as csvfile: 
    writer = csv.writer(csvfile)
    # 多行寫入用writerows
    writer.writerows(data_array)
    # 單行逐個寫入用 writerow
#     writer.writerow(list1)
#     writer.writerow(list2)
    
    # 執行新增資料操作之後,要寫close關閉,否則下次無法再次插入新的資料
    csvfile.close()

實時寫入資料時,有可能是逐個寫入,也可能是一次性寫入多個資料。多行寫入用writerows,

單行逐個寫入用 writerow,根據需求調整。close()這行程式碼一定要加上,否則下次無法再次插入新的資料。

執行結果:

再次逐個寫入,程式碼:

list1=[0,0,0]
list2=[1,1,1]
data_array=[[5,5,5],[1,2,3]]
with open("test3.csv","a",newline='') as csvfile: 
    writer = csv.writer(csvfile)
    # 寫入多行用writerows
#     writer.writerows(data_array)
#     寫入單行用 writerow
    writer.writerow(list1)
    writer.writerow(list2)
    
    # 執行新增資料操作之後,要寫close關閉,否則下次無法再次插入新的資料
    csvfile.close()

執行結果:

結果顯示,逐個新增資料成功。

注意的地方

如果不是逐行寫入,而是直接將陣列一次性寫入到csv檔案中(相當於多行寫入),則上述程式碼中改用writerows即可

2:實時向txt檔案寫入資料

實時向txt檔案寫入內容的過程,與建立csv檔案,實時向檔案寫入內容大致相同,只需要新增一個換行符就行。

程式碼:

with open ('testing.txt','a') as f:
    f.write('%s       %s      %s'%('姓名','國籍','金額'))

執行結果:

結果生成了一個testing.txt的檔案。

再次向txt檔案中寫入資料

程式碼:

with open ('testing.txt','a') as f:   
    f.write('n')     #換行
    f.write('%s       %s      %d' %('張三','中國',2000))

執行結果:

結果顯示,已經將資料寫入txt檔案中。

以上為個人經驗,希望能給大家一個參考,也希望大家多多支援it145.com。


IT145.com E-mail:sddin#qq.com