在Python中進行CSV文件的讀取與寫入操作

更新時間：2025年03月09日 15:16:02 作者：climber1121

本文詳細介紹了如何在Python中使用csv模塊進行CSV文件的讀取和寫入,包括基本用法和高級功能,如DictWriter和DictReader,以及注意事項,需要的朋友可以參考下

1 引言

在數(shù)據(jù)處理和數(shù)據(jù)分析領域，CSV (逗號分隔值) 文件是一種常見的文件格式，用于存儲表格數(shù)據(jù)。Python 通過內(nèi)置的 csv 模塊提供了對 CSV 文件的讀寫支持，使得處理這種類型的文件變得簡單高效。本文將詳細介紹如何在 Python 中進行 CSV 文件的讀取和寫入操作。

2 CSV 文件簡介

CSV 文件是一種簡單的文本文件，其中的數(shù)據(jù)由逗號分隔。每行代表一個數(shù)據(jù)記錄，每個記錄可以包含多個字段。CSV 文件被廣泛用于數(shù)據(jù)交換和數(shù)據(jù)存儲。

3 Python 中的 csv 模塊

Python 的 csv 模塊提供了一系列函數(shù)來讀取和寫入 CSV 文件。它可以處理各種類型的 CSV 數(shù)據(jù)，包括不同的分隔符和引號約定。

4 寫入 CSV 文件

4.1 基本用法

import csv

# 數(shù)據(jù)
data = [
    ['Name', 'Age', 'City'],
    ['Alice', 30, 'New York'],
    ['Bob', 25, 'Los Angeles']
]

# 寫入 CSV 文件
with open('example.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

4.2 高級用法

使用 DictWriter 寫入 CSV，提供了字段名的映射。

with open('example.csv', 'w', newline='') as file:
    fields = ['Name', 'Age', 'City']
    writer = csv.DictWriter(file, fieldnames=fields)

    writer.writeheader()
    writer.writerow({'Name': 'Alice', 'Age': 30, 'City': 'New York'})
    writer.writerow({'Name': 'Bob', 'Age': 25, 'City': 'Los Angeles'})

5 讀取 CSV 文件

5.1 基本用法

# 讀取 CSV 文件
with open('example.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

5.2 高級用法

使用 DictReader 讀取 CSV，每行數(shù)據(jù)被映射到一個字典。

with open('example.csv', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row)

6 實例演示

以下是一個 Python 示例，演示如何使用 csv 模塊來讀取和寫入 CSV 文件。這個示例將包含兩個主要部分：一個用于寫入 CSV 文件的函數(shù)和另一個用于讀取 CSV 文件的函數(shù)。我們將這些功能封裝在一個名為 csv_example.py 的 Python 文件中。

# csv_example.py
import csv

def write_csv(file_name, data, field_names=None):
    """ 寫入 CSV 文件 """
    with open(file_name, 'w', newline='', encoding='utf-8') as file:
        if field_names:
            writer = csv.DictWriter(file, fieldnames=field_names)
            writer.writeheader()
            writer.writerows(data)
        else:
            writer = csv.writer(file)
            writer.writerows(data)

def read_csv(file_name):
    """ 讀取 CSV 文件 """
    with open(file_name, 'r', newline='', encoding='utf-8') as file:
        reader = csv.reader(file)
        for row in reader:
            print(row)

def main():
    # CSV 文件名
    file_name = 'example.csv'

    # 數(shù)據(jù) - 字典格式
    data_dict = [
        {'Name': 'Alice', 'Age': 30, 'City': 'New York'},
        {'Name': 'Bob', 'Age': 25, 'City': 'Los Angeles'}
    ]

    # 數(shù)據(jù) - 列表格式
    data_list = [
        ['Name', 'Age', 'City'],
        ['Alice', 30, 'New York'],
        ['Bob', 25, 'Los Angeles']
    ]

    # 寫入 CSV 文件
    write_csv(file_name, data_dict, field_names=['Name', 'Age', 'City'])
    # 或者使用列表數(shù)據(jù)
    # write_csv(file_name, data_list)

    # 讀取 CSV 文件
    read_csv(file_name)

if __name__ == '__main__':
    main()

運行以上代碼控制臺輸出結果：[‘Name’, ‘Age’, ‘City’]
[‘Alice’, ‘30’, ‘New York’]
[‘Bob’, ‘25’, ‘Los Angeles’]
并生成文件：“example.csv”，內(nèi)容如下：

這個腳本定義了兩個函數(shù)：write_csv 和 read_csv。write_csv 函數(shù)可以根據(jù)傳入的數(shù)據(jù)格式（字典或列表）將數(shù)據(jù)寫入 CSV 文件。read_csv 函數(shù)則用于讀取 CSV 文件并打印其內(nèi)容。main 函數(shù)中整合了這些操作的流程。運行這個腳本將會創(chuàng)建一個名為 example.csv 的文件，并在其中寫入數(shù)據(jù)，然后讀取并打印出這些數(shù)據(jù)。