Kaedah dan teknik yang cekap untuk memproses fail CSV dalam Python

Python menyediakan modul csv yang cekap untuk memproses fail CSV. Anda boleh melakukan ini dengan mengikuti langkah berikut: Pasang modul csv. Gunakan csv.reader() untuk membaca fail CSV. Memproses fail besar baris demi baris dengan cekap. Baca dan tulis fail CSV dalam memori untuk pemprosesan yang lebih pantas. Gunakan csv.writer() untuk menulis fail CSV. Memproses fail CSV dalam format yang berbeza melalui dialek. Sesuaikan pembatas dan petikan untuk membaca fail CSV bukan standard.

Kaedah dan teknik yang cekap untuk memproses fail CSV dalam Python

Kes praktikal: membaca dan menganalisis fail CSV data jualan

csv 模块

使用 Python 的 pip 安装 csv 模块：

pip install csv

2. 读取 CSV 文件

使用 csv.reader() 读取 CSV 文件：

import csv

with open('sales_data.csv', 'r') as csv_file:
    csv_reader = csv.reader(csv_file)
    for row in csv_reader:
        print(row)  # 打印每一行数据

3. 逐行处理

使用逐行处理生成器，更高效地处理大文件：

import csv

with open('sales_data.csv', 'r') as csv_file:
    csv_reader = csv.reader(csv_file)
    for row in csv_reader:
        process_row(row)  # 调用自定义函数处理每一行

4. 读写内存中的 CSV

使用 StringIO 类将 CSV 数据加载到内存中，便于快速处理：

import csv
from io import StringIO

csv_data = """
1,John Doe,100
2,Jane Smith,200
"""

csv_file = StringIO(csv_data)
csv_reader = csv.reader(csv_file)
for row in csv_reader:
    print(row)

5. 写出 CSV 文件

使用 csv.writer()1 Gunakan pip Python memasang modul csv:

import csv

with open('output.csv', 'w', newline='') as csv_file:
    csv_writer = csv.writer(csv_file)
    csv_writer.writerow(['ID', 'Name', 'Sales'])
    csv_writer.writerow([1, 'John Doe', 100])

2 Baca fail CSV

Gunakan csv.reader() untuk membaca CSV. Fail:

import csv

with open('sales_data.csv', 'r') as csv_file:
    csv_reader = csv.reader(csv_file, dialect='excel')
    for row in csv_reader:
        print(row)

3. Pemprosesan baris demi baris

Gunakan penjana pemprosesan baris demi baris untuk memproses fail besar dengan lebih cekap:

import csv

with open('custom_data.csv', 'r') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=';', quotechar='"')
    for row in csv_reader:
        print(row)

Atas ialah kandungan terperinci Kaedah dan teknik yang cekap untuk memproses fail CSV dalam Python. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!