python怎么處理csv文件的數據

198次閱讀

共計 1107 個字符，預計需要花費 3 分鐘才能閱讀完成。

在 Python 中，可以使用內置的 csv 模塊來處理 CSV 文件的數據。下面是一些常見的操作：

讀取 CSV 文件：使用 csv.reader 函數來讀取 CSV 文件中的數據。例如，假設有一個名為 data.csv 的 CSV 文件，可以使用以下代碼來讀取它的數據：

import csv

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

寫入 CSV 文件：使用 csv.writer 函數來寫入數據到 CSV 文件中。例如，假設要將一個名為 data 的列表寫入到一個名為 output.csv 的 CSV 文件中，可以使用以下代碼：

import csv

data = [['Name', 'Age', 'Country'],
        ['John', 25, 'USA'],
        ['Alice', 30, 'Canada']]

with open('output.csv', 'w', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(data)

處理 CSV 文件的每一行數據：使用 csv.reader 函數讀取 CSV 文件的每一行數據后，可以使用循環來處理每一行的數據。例如，可以將每行數據存儲到一個列表中：

import csv

with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    data = []
    for row in csv_reader:
        data.append(row)

# 處理 data 中的數據

索引和訪問 CSV 數據：可以使用索引來訪問 CSV 文件中的特定數據。例如，假設 CSV 文件的第一行是標題行，可以使用索引來訪問特定列的數據：

import csv

with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        name = row[0]
        age = row[1]
        country = row[2]
        print(name, age, country)