1.csv文件介紹

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

2.寫(xiě)入CSV文件

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

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

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

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

import csv
# 創(chuàng)建要寫(xiě)入的數(shù)據(jù)
data = [
    ['張三', 20, '男'],
    ['李四', 25, '女'],
    ['王五', 30, '男'],
    ['"Tom, Jerry"', 35, '男']
]
# 打開(kāi)文件，指定寫(xiě)入模式
with open('data.csv', 'w', newline='') as file:
    # 創(chuàng)建csv寫(xiě)入對(duì)象
    writer = csv.writer(file)
    # 寫(xiě)入表頭
    writer.writerow(['姓名', '年齡', '性別'])
    # 寫(xiě)入數(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)。為了寫(xiě)入這個(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ù)就可以正確解析了。

總結(jié)

到此這篇關(guān)于Python中csv文件的寫(xiě)入與讀取的文章就介紹到這了,更多相關(guān)Python csv文件寫(xiě)入與讀取內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論