Python對csv文件的讀寫操作

python內置了csv模塊，用它可以方便的操作csv文件。

1、寫文件

（1）寫文件的方法一

import csv

# open 打開文件有多種模式，下面是常見的4種
# r：讀數據，默認模式
# w：寫數據，如果已有數據則會先清空
# a：向文件末尾追加數據
# x : 寫數據，如果文件已存在則失敗
# 第2至4種模式如果第一個參數指定的文件不存在，則會先創建一個空文件
with open('1.csv', 'w', newline='') as f:    
    head = ['標題列1', '標題列2']
    rows = [
                ['張三', 80],
                ['李四', 90]
            ]  
    writer = csv.writer(f) 
    #寫入一行數據
    writer.writerow(head) 
    #寫入多行數據
    writer.writerows(rows)  

（2）寫文件的方法二

import csv

with open('1.csv', 'w', newline='') as f:    
    head = ['標題列1', '標題列2']
    rows = [
                {'標題列1': '張三', '標題列2' :80},
                {'標題列1': '李四', '標題列2' :90}
            ]
    writer = csv.DictWriter(f,head)
    writer.writeheader()
    writer.writerows(rows)

2、讀文件

以上面創建的1.csv為例讀文件

import csv

with open('1.csv') as f:    
    reader = csv.reader(f)
    for row in reader:
        print(row)

運行結果：

['標題列1', '標題列2']
['張三', '80']
['李四', '90']