快捷導(dǎo)航

Python中csv文件的寫入與讀取方法例子

更新時(shí)間：2023年09月19日 08:27:26 作者：初見(jiàn)～

這篇文章主要給大家介紹了關(guān)于Python中csv文件的寫入與讀取方法的相關(guān)資料,csv是"Comma-Separated Values(逗號(hào)分割的值)"的首字母縮寫,它其實(shí)和txt文件一樣,都是純文本文件,使用Python來(lái)讀寫csv文件是非常容易的,需要的朋友可以參考下

1.csv文件介紹

CSV文件是一種常見(jiàn)的數(shù)據(jù)格式，它以逗號(hào)分隔不同的字段，每行表示一個(gè)數(shù)據(jù)記錄。在Python中，我們可以使用csv模塊來(lái)讀取和寫入CSV文件。

2.寫入CSV文件

在Python中，我們可以使用csv模塊的writer對(duì)象來(lái)寫入CSV文件。下面是一個(gè)例子：

import csv
# 創(chuàng)建要寫入的數(shù)據(jù)
data = [
    ['張三', 20, '男'],
    ['李四', 25, '女'],
    ['王五', 30, '男']
]
# 打開(kāi)文件，指定寫入模式
with open('data.csv', 'w', newline='') as file:
    # 創(chuàng)建csv寫入對(duì)象
    writer = csv.writer(file)
    # 寫入表頭
    writer.writerow(['姓名', '年齡', '性別'])
    # 寫入數(shù)據(jù)
    writer.writerows(data)

在上面的例子中，我們首先創(chuàng)建了要寫入的數(shù)據(jù)，它是一個(gè)包含多個(gè)列表的列表。然后使用open()函數(shù)打開(kāi)文件，并指定寫入模式。newline=''表示不使用額外的換行符。接著創(chuàng)建了一個(gè)csv寫入對(duì)象writer，使用writerow()方法寫入表頭，使用writerows()方法寫入數(shù)據(jù)。最后，使用with語(yǔ)句來(lái)自動(dòng)關(guān)閉文件。

需要注意的是，寫入CSV文件時(shí)，需要將數(shù)據(jù)轉(zhuǎn)換成字符串格式。如果寫入的數(shù)據(jù)中包含逗號(hào)、雙引號(hào)等特殊字符，需要進(jìn)行處理。例如：

import csv
# 創(chuàng)建要寫入的數(shù)據(jù)
data = [
    ['張三', 20, '男'],
    ['李四', 25, '女'],
    ['王五', 30, '男'],
    ['"Tom, Jerry"', 35, '男']
]
# 打開(kāi)文件，指定寫入模式
with open('data.csv', 'w', newline='') as file:
    # 創(chuàng)建csv寫入對(duì)象
    writer = csv.writer(file)
    # 寫入表頭
    writer.writerow(['姓名', '年齡', '性別'])
    # 寫入數(shù)據(jù)
    for row in data:
        # 將包含特殊字符的數(shù)據(jù)進(jìn)行處理
        row[0] = row[0].replace('"', '""')
        row[0] = f'"{row[0]}"'
        writer.writerow(row)

在上面的例子中，我們創(chuàng)建了一個(gè)包含特殊字符的數(shù)據(jù)，它包含逗號(hào)和雙引號(hào)。為了寫入這個(gè)數(shù)據(jù)，我們需要將雙引號(hào)進(jìn)行轉(zhuǎn)義，并在數(shù)據(jù)前后加上雙引號(hào)。具體來(lái)說(shuō)，我們使用replace()方法將雙引號(hào)替換成兩個(gè)雙引號(hào)，然后使用f-string將數(shù)據(jù)前后加上雙引號(hào)。

3.讀取CSV文件

在Python中，我們可以使用csv模塊的reader對(duì)象來(lái)讀取CSV文件。下面是一個(gè)例子：

import csv
# 打開(kāi)文件，指定讀取模式
with open('data.csv', 'r') as file:
    # 創(chuàng)建csv讀取對(duì)象
    reader = csv.reader(file)
    # 遍歷每一行數(shù)據(jù)
    for row in reader:
        print(row)

在上面的例子中，我們使用open()函數(shù)打開(kāi)文件，并指定讀取模式。然后創(chuàng)建了一個(gè)csv讀取對(duì)象reader，使用for循環(huán)遍歷每一行數(shù)據(jù)，并使用print()函數(shù)輸出。

需要注意的是，在讀取CSV文件時(shí)，每一行數(shù)據(jù)都會(huì)被解析成一個(gè)列表。如果CSV文件中有表頭，第一行數(shù)據(jù)就是表頭，可以使用next()函數(shù)跳過(guò)：

import csv
# 打開(kāi)文件，指定讀取模式
with open('data.csv', 'r') as file:
    # 創(chuàng)建csv讀取對(duì)象
    reader = csv.reader(file)
    # 跳過(guò)表頭
    next(reader)
    # 遍歷每一行數(shù)據(jù)
    for row in reader:
        print(row)

在上面的例子中，我們使用next()函數(shù)跳過(guò)了表頭。

另外，如果CSV文件中包含特殊字符，例如逗號(hào)、雙引號(hào)等，需要使用csv.reader的quotechar參數(shù)來(lái)指定轉(zhuǎn)義字符。例如：

import csv
# 打開(kāi)文件，指定讀取模式
with open('data.csv', 'r') as file:
    # 創(chuàng)建csv讀取對(duì)象
    reader = csv.reader(file, quotechar='"')
    # 遍歷每一行數(shù)據(jù)
    for row in reader:
        print(row)

在上面的例子中，我們使用quotechar='"'來(lái)指定雙引號(hào)為轉(zhuǎn)義字符。這樣，包含雙引號(hào)的數(shù)據(jù)就可以正確解析了。