使用Python創建CSV文件並寫入數據

本文將介紹如何使用Python創建CSV文件並寫入數據，主要包括以下幾個方面：

一、CSV文件介紹

CSV(Comma-Separated Values)文件是一種常用的簡單數據交換格式，通常用於把表格數據從一種應用程序轉移到另一種應用程序。CSV文件由文本行組成，每行表示一個數據記錄，每個字段用逗號分隔。CSV文件可以使用任何文本編輯器打開，很容易存儲和處理。

二、創建CSV文件

Python內置的csv模塊提供了一些方法來創建CSV文件，其中最常用的是csv.writer()。csv.writer()將數據寫入文件對象，並自動加上逗號和換行符。以下是創建CSV文件的簡單示例：

import csv

with open('example.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile, delimiter=',')
    writer.writerow(['Name', 'Age', 'City'])
    writer.writerow(['Tom', '28', 'Beijing'])
    writer.writerow(['Jack', '30', 'Shanghai'])

以上代碼創建了一個名為example.csv的CSV文件，寫入了三個數據記錄，每個記錄包括Name、Age和City三個字段。寫入文件的過程使用了with語句，該語句可以自動關閉文件，確保寫入數據的完整性。

三、讀取CSV文件

除了創建CSV文件外，Python的csv模塊還提供了一些方法來讀取CSV文件。csv.reader()方法可以將CSV文件轉換為Python的列表或者字典形式，方便後續的數據處理和分析。以下是讀取CSV文件的示例：

import csv

with open('example.csv', newline='') as csvfile:
    reader = csv.reader(csvfile, delimiter=',')
    for row in reader:
        print(', '.join(row))

以上代碼使用csv.reader()方法讀取了example.csv文件並逐行打印。csv.reader()的delimiter參數指定了字段分隔符，默認為’,’。

四、寫入字典數據

除了寫入列表數據，我們還可以將字典數據寫入CSV文件。以下是寫入字典數據的示例：

import csv

with open('example_dict.csv', 'w', newline='') as csvfile:
    fieldnames = ['Name', 'Age', 'City']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerow({'Name': 'Tom', 'Age': '28', 'City': 'Beijing'})
    writer.writerow({'Name': 'Jack', 'Age': '30', 'City': 'Shanghai'})

以上代碼創建了一個名為example_dict.csv的CSV文件，寫入了兩個數據記錄，每個記錄包括Name、Age和City三個字段。寫入文件的過程使用了DictWriter類，它允許我們通過字典的形式來寫入CSV內容。需要特別注意的是，在寫入數據前需要使用writeheader()方法寫入表頭信息。

五、讀取字典數據

和寫入字典數據類似，我們也可以使用csv.DictReader()方法來讀取CSV文件中的字典數據。以下是讀取字典數據的示例：

import csv

with open('example_dict.csv', newline='') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        print(row['Name'], row['Age'], row['City'])

以上代碼使用csv.DictReader()方法讀取了example_dict.csv文件並逐行打印。csv.DictReader()方法將每一行數據轉換為一個字典，其中字典的鍵對應於表頭信息，字典的值對應於每個字段的值。

六、總結

本文介紹了如何使用Python創建CSV文件並寫入數據，以及讀取CSV文件的方法。通過學習本文，你可以使用Python輕鬆地進行CSV文件數據的讀寫和處理，實現數據的快速轉換和分析。