Python 實時向文件寫入數據(附代碼)
之前在做數據分析的過程中,需要對數據進行實時的寫入,比如對新生成的數據寫入之前已經生成的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文件中。