使用Python導出CSV文件

介紹

CSV（Comma-Separated Values）是一種常見的數據格式，以逗號作為數據值之間的分隔符，在數據倉庫和其他應用程序之間共享數據。Python作為一種通用的編程語言，也可以用於處理CSV文件。

本文將介紹如何使用Python導出CSV文件，以及如何使用Python在CSV文件中讀取、操作和寫入數據。

正文

一、導出CSV文件

使用Python導出CSV文件非常簡單。下面是一個簡單的示例：

import csv

with open('example.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age', 'Gender'])
    writer.writerow(['John Smith', '33', 'Male'])
    writer.writerow(['Jane Doe', '28', 'Female'])

運行上述代碼，將在當前目錄下創建名為example.csv的新文件，其內容如下：

Name,Age,Gender
John Smith,33,Male
Jane Doe,28,Female

代碼說明：

• csv.writer：使用csv.writer函數創建一個寫入器實例。
• writer.writerow：向CSV文件中寫入一行數據。
• mode=’w’：打開文件用於寫入，如果文件不存在，則創建該文件。
• newline=”：防止CSV文件中出現空行。

二、從CSV文件中讀取數據

Python也可以使用csv模塊讀取CSV文件中的數據。下面是一個讀取CSV文件的示例：

import csv

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

運行上述代碼，將輸出如下內容：

['Name', 'Age', 'Gender']
['John Smith', '33', 'Male']
['Jane Doe', '28', 'Female']

代碼說明：

• csv.reader：使用csv.reader函數創建一個讀取器實例。
• for row in reader：遍歷整個CSV文件，並列印每一行數據。
• mode=’r’：打開文件用於讀取。

三、操作CSV文件中的數據

Python中可以使用csv模塊對CSV文件中的數據進行操作。下面是一些示例：

1. 使用列表推導式對CSV文件中的數據進行篩選

示例代碼如下：

import csv

with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    male_rows = [row for row in reader if row[2] == 'Male']
print(male_rows)

運行上述代碼，將輸出CSV文件中所有性別為Male的行：

[['John Smith', '33', 'Male']]

代碼說明：

• [row for row in reader if row[2] == ‘Male’]：使用列表推導式篩選出所有性別為Male的行。

2. 對CSV文件中的數據進行排序

示例代碼如下：

import csv

with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    sorted_rows = sorted(reader, key=lambda row: row[1])
print(sorted_rows)

運行上述代碼，將按照CSV文件中Age列進行升序排序，並輸出所有行：

[['Jane Doe', '28', 'Female'], ['John Smith', '33', 'Male']]

代碼說明：

• sorted(reader, key=lambda row: row[1])：使用lambda函數進行升序排序，排序關鍵字為Age列。

3. 使用字典進行CSV文件中數據的操作

示例代碼如下：

import csv

with open('example.csv', mode='r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row['Name'], row['Age'], row['Gender'])

運行上述代碼，將輸出所有行的Name、Age和Gender列的值：

John Smith 33 Male
Jane Doe 28 Female

代碼說明：

• csv.DictReader：使用csv.DictReader函數創建一個字典讀取器實例。
• row[‘Name’]、row[‘Age’]、row[‘Gender’]：使用字典的方式獲取CSV文件中每一行的數據。

小結

本文介紹了如何使用Python導出CSV文件、從CSV文件中讀取數據以及操作CSV文件中的數據。CSV文件是一種常見的數據格式，在數據處理、數據倉庫和其他應用程序之間使用廣泛，因此了解如何使用Python處理CSV文件是非常有用的。