Python查詢CSV文件的全面指南

本篇文章將從多個方面，詳細闡述如何使用Python查詢CSV文件。無論您是初學者還是有經驗的開發人員，您都可以從本文中獲取有益的信息。

一、CSV文件介紹

CSV文件是一種純文本文件，可以用來存儲表格數據。在CSV文件中，每一行都代表了表格中的一行數據，每一列則代表了表格中的一列數據。每個單元格以逗號作為分隔符。

CSV文件的優點是易於閱讀和編輯。另外，它們可以通過各種各樣的軟體包（如Microsoft Excel，Google Sheets和LibreOffice Calc）進行打開和修改。這使得CSV文件成為非常流行的數據交換格式。

二、CSV查詢基礎

在Python中，我們可以使用csv模塊來讀取和寫入CSV文件。下面是一個簡單的例子：

import csv

# 打開文件
with open('data.csv', 'r') as file:
    # 創建reader對象
    reader = csv.reader(file)

    # 遍歷行
    for row in reader:
        # 遍歷單元格
        for cell in row:
            print(cell)

上面的代碼使用csv模塊中的reader方法來讀取data.csv文件。在這個例子中，我們遍歷了CSV文件的每一行，並列印出每個單元格的值。

三、查詢特定數據

在CSV文件中查詢特定的數據需要一些額外的代碼。下面的代碼將演示如何查詢一列具有特定值的數據：

import csv

# 打開文件
with open('data.csv', 'r') as file:
    # 創建reader對象
    reader = csv.reader(file)

    # 遍歷行
    for row in reader:
        # 如果第一列包含"apple"，則列印該行
        if row[0] == "apple":
            print(row)

上面的代碼將列印出包含單元格第一列為「apple」的所有行。

我們還可以在CSV文件中查詢多個條件，例如查詢單元格第一列為「apple」且單元格第二列為「3」的所有行：

import csv

# 打開文件
with open('data.csv', 'r') as file:
    # 創建reader對象
    reader = csv.reader(file)

    # 遍歷行
    for row in reader:
        # 如果第一列包含"apple"且第二列包含"3"，則列印該行
        if row[0] == "apple" and row[1] == "3":
            print(row)

四、CSV數據處理

在某些情況下，我們可能需要對CSV數據進行處理。例如，我們可能需要計算CSV文件中每列的平均值，或者我們可能需要將字元串轉換為數字。

下面的代碼將演示如何將字元串列錶轉換為整數列表：

import csv

# 打開文件
with open('data.csv', 'r') as file:
    # 創建reader對象
    reader = csv.reader(file)

    # 遍歷行
    for row in reader:
        # 將字元串列錶轉換為整數列表
        int_list = [int(i) for i in row]

        # 計算整數列表的總和
        sum_num = sum(int_list)

        # 列印結果
        print(sum_num)

上面的代碼將列印出每行中所有數字之和。

五、CSV數據寫入

在Python中，我們可以使用csv模塊將數據寫入CSV文件。下面的代碼演示如何將列表寫入CSV文件：

import csv

# 要寫入CSV的數據
data = [
    ['apple', '3', '1.2'],
    ['orange', '4', '1.3'],
    ['banana', '2', '1.1']
]

# 打開文件並寫入數據
with open('output.csv', 'w') as file:
    # 創建writer對象
    writer = csv.writer(file)

    # 寫入數據
    for row in data:
        writer.writerow(row)

上面的代碼將創建一個名為output.csv的新文件，並將數據寫入其中。

結論

本篇文章展示了如何使用Python查詢CSV文件。使用csv模塊，我們可以輕鬆地讀取、處理和寫入CSV文件。希望這篇文章對您有所幫助！