Python讀寫csv文件的操作方法

這篇文章主要介紹了Python讀寫csv文件的操作方法，具有一定借鑒價值，需要的朋友可以參考下。下面就和我一起來看看吧。

要在 Python 中寫入 CSV，請使用 Python 的 csv 模塊。

例如，讓我們將一個字符串列表寫入一個新的 CSV 文件：

import csv
data = ["This", "is", "a", "Test"]
with open('example.csv', 'w') as file:
    writer = csv.writer(file)
    writer.writerow(data)

 因此，您會在當前文件夾中看到一個名為 example.csv 的文件。

用 Python 編寫 CSV 的 4 個步驟

要在 Python 中寫入 CSV 文件：

1. 以寫入模式打開 CSV 文件。 這是使用 open() 函數發生的。 給它文件的路徑作為第一個參數。 將模式指定為第二個參數（“r”表示讀取，“w”表示寫入）。
2. 創建 CSV 編寫器對象。 為此，創建一個 csv 模塊的 writer() 對象，并將打開的文件作為其參數傳遞。
3. 將數據寫入 CSV 文件。 使用 writer 對象的 writerow() 函數將數據寫入 CSV 文件。
4. 關閉 CSV 文件，使用文件的 close() 方法。

這是一個說明此過程的示例：

import csv
# 1.
file = open('test.csv', 'w')
# 2.
writer = csv.writer(file)
# 3.
data = ["This", "is", "a", "Test"]
writer.writerow(data)
# 4.
file.close()

這段代碼在當前文件夾中創建了一個名為 test.csv 的文件。

如果指定的文件不存在，open() 函數將打開一個新文件。 如果是，則打開現有文件。

速寫

要縮短寫入 CSV 的時間，請使用 with 語句打開文件。 這樣你就不用擔心自己關閉文件了。 with 會自動處理該部分。

例如：

import csv
# 1. step
with open('test.csv', 'w') as file:
    # 2. step
    writer = csv.writer(file)
    # 3. step
    data = ["This", "is", "a", "Test"]
    writer.writerow(data)

這將在當前文件夾中創建一個名為 test.csv 的新 CSV 文件，并將字符串列表寫入其中。

如何在 Python 中將非 ASCII 字符寫入 CSV

默認情況下，您不能將非 ASCII 字符寫入 CSV 文件。

要支持將非 ASCII 值寫入 CSV 文件，請在 open() 調用中將字符編碼指定為第三個參數。

with open('PATH_TO_FILE.csv', 'w', encoding="UTF8")

其余過程遵循您之前學到的步驟。

如何為 CSV 文件創建標題

到目前為止，您已經創建了缺少結構的 CSV 文件。

在 Python 中，可以使用用于將任何數據寫入 CSV 

writerow() 函數為任何 CSV 文件編寫標頭。

 例子:  讓我們創建一個包含學生數據的示例 CSV 文件。

為了很好地構建數據，為學生創建一個標題并將其插入 CSV 文件的開頭。 在此之后，您可以按照之前的相同步驟將數據寫入 CSV 文件。

這是代碼：

import csv
# Define the structure of the data
student_header = ['name', 'age', 'major', 'minor']
# Define the actual data
student_data = ['Jack', 23, 'Physics', 'Chemistry']
# 1. Open a new CSV file
with open('students.csv', 'w') as file:
    # 2. Create a CSV writer
    writer = csv.writer(file)
    # 3. Write data to the file
    writer.writerow(student_header)
    writer.writerow(student_data)

這會將 students.csv 文件創建到您當前正在使用的文件夾中。新文件如下所示：

如何在 Python 中將多行寫入 CSV 文件

在 Python 中，您可以使用 CSV 編寫器的 writerows() 函數同時將多行寫入 CSV 文件。

例子。 假設您要將多行數據寫入 CSV 文件。 例如，您可能有一個學生列表，而不是只有其中一個。

要將多行數據寫入 CSV，請使用 writerows() 方法。

這是一個例子：

import csv
student_header = ['name', 'age', 'major', 'minor']
student_data = [
    ['Jack', 23, 'Physics', 'Chemistry'],
    ['Sophie', 22, 'Physics', 'Computer Science'],
    ['John', 24, 'Mathematics', 'Physics'],
    ['Jane', 30, 'Chemistry', 'Physics']
]
with open('students.csv', 'w') as file:
    writer = csv.writer(file)
    writer.writerow(student_header)
    # Use writerows() not writerow()
    writer.writerows(student_data)

 這會生成一個新的 CSV 文件，如下所示：

如何在 Python 中將字典寫入 CSV 文件

要在 Python 中將字典寫入 CSV 文件，請按照以下三個步驟使用 DictWriter 對象：

1. 使用 csv 模塊的 DictWriter 對象并在其中指定字段名稱。

2. 使用 writeheader() 方法將標頭創建到 CSV 文件中。

3. 使用 writerows() 方法將字典數據寫入文件。

例子：讓我們將學生數據字典寫入 CSV 文件。

import csv
student_header = ['name', 'age', 'major', 'minor']
student_data = [
    {'name': 'Jack', 'age': 23, 'major': 'Physics', 'minor': 'Chemistry'},
    {'name': 'Sophie', 'age': 22, 'major': 'Physics', 'minor': 'Computer Science'},
    {'name': 'John', 'age': 24, 'major': 'Mathematics', 'minor': 'Physics'},
    {'name': 'Jane', 'age': 30, 'major': 'Chemistry', 'minor': 'Physics'}
]
with open('students.csv', 'w') as file:
    # Create a CSV dictionary writer and add the student header as field names
    writer = csv.DictWriter(file, fieldnames=student_header)
    # Use writerows() not writerow()
    writer.writeheader()
    writer.writerows(student_data)

現在結果與前面示例中的 students.csv 文件相同：

結論

CSV 或逗號分隔值是一種常用的文件格式。 它由通常用逗號分隔的值組成。

要在 Python 中寫入 CSV，您需要通過以下步驟使用 csv 模塊：

1. 以寫入模式打開 CSV 文件。

2. 創建 CSV 編寫器對象。

3. 將數據寫入 CSV 文件。

4. 關閉 CSV 文件。

這是一個實際的例子。

import csv
data = ["This", "is", "a", "Test"]
with open('example.csv', 'w') as file:
    writer = csv.writer(file)
    writer.writerow(data)

編碼愉快！

以上就是Python讀寫csv文件的操作方法的詳細內容了，看完之后是否有所收獲呢？如果想了解更多相關內容，歡迎來億速云行業資訊！