Python操作csv文件之csv.writer()和csv.DictWriter()方法的基本使用

一、csv.writer()方法的基本使用

示例代碼1:

import csv
data = [
    (1,"cat",18),
    (2,"dog",17),
    (3,"tigger",16),
]
f = open('test.csv','a',encoding='utf8',newline='')
writer = csv.writer(f)  # csv.writer()中可以傳一個文件對象
for line in data: # 該data既可以是列表嵌套列表的數據類型也可以是列表嵌套元組的數據類型
    writer.writerow(line)

writer()的功能是創建一個常規Writer的對象,但是調用writer()的writerow/writerows方法是要傳入的是列表類型數據。

註意示例代碼1中打開文件時f = open('test.csv','a',encoding='utf8',newline='')指定newline=‘’參數可以使每次寫入數據不會產生空行,註意根據需求指定文件的打開方式,該案例中是使用追加的方式打開文件,這樣寫入數據的時候,就不會產生是數據的覆蓋。

代碼成功運行在test.csv中產生的結果為:

運行結果

二、csv.DictWriter()方法的基本使用

示例代碼2:

import csv
data = [
    {'id':1,'name':'dog',"age":18},
    {'id':2,'name':'cat',"age":19},
    {'id':3,'name':'dog',"age":20},
]
f = open('test1.csv','a',encoding='utf8',newline='')  # 指定newline=‘'參數
writer = csv.DictWriter(f,fieldnames=['id','name','age'])
writer.writeheader() # 將字段寫入csv格式文件首行
for line in data:
    writer.writerow(line)

DictWriter()的功能是創建一個類似於常規Writer的對象,但是調用DictWriter()的writerow/writerows方法是要傳入的是字典類型數據。

示例代碼2中的writer.writeheader()作用是將字段寫入,即將DictWriter構造方法的fieldnames參數中的字段寫入csv格式文件的首行,

如果未執行writeheader()方法的話是不會將fieldnames中的字段寫入csv格式文件的首行。

代碼成功運行在test1.csv中產生的結果為:

運行結果

總結 

到此這篇關於Python操作csv文件之csv.writer()和csv.DictWriter()方法基本使用的文章就介紹到這瞭,更多相關Python csv.writer() csv.DictWriter()方法內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: