用Python编写CSV文件的教程

256 阅读1分钟

在本教程中,你将学习如何用Python编写CSV文件。

CSV(逗号分隔的值)是用于存储和读取数据的流行文件类型。所以在本教程中,我们将使用buildcsv 模块来创建和写入数据到CSV文件。

因此,让我们继续进行编码。

编写CSV文件的步骤

首先,我们将从Python中导入内置的CSV 模块。

import csv

然后我们将在w 写入模式下打开文件,使用open() 函数写入文件。

file = open('path/csv_file', 'w')

然后我们通过调用csv 模块中的writer() 函数,用文件对象创建一个CSV写作者对象。

writer = csv.writer(file)

然后,我们将通过调用writerow()writerows() 方法写入对象将数据写入CSV文件中。

如果写单行,我们可以使用writerow() 方法。

writer.writerows(columnValues)

如果想写多行,我们需要使用writerows() 方法。

writer.writerows(columnValues)

最后,通过调用close() 方法关闭打开的文件。

file.close()

写入单行到CSV文件

下面是创建一个CSV文件并写入单行数据的完整代码。

# write.py

import csv

columnName = ['name', 'age', 'country']

columnValues = ['Any', 44, 'Germany']

with open('emp.csv', 'w', encoding='UTF8', newline='') as file:

    writer = csv.writer(file)

    # write column names
    writer.writerow(columnName)

    # write rows values
    writer.writerow(columnValues)

向CSV文件写入多行数据

下面是创建一个CSV文件并写入多行数据的完整代码。

# write.py

import csv

columnName = ['name', 'age', 'country']

columnValues = [
    ['Smith', 30, 'Australia'],
    ['William', 40, 'Newyork'],
    ['Andre', 60, 'Germany'],
    ['Roger', 55, 'Paris'],
    ['Donal', 33, 'Sydney']   
]

with open('employee.csv', 'w', encoding='UTF8', newline='') as file:

    writer = csv.writer(file)

    # write column names
    writer.writerow(columnName)

    # write rows values
    writer.writerows(columnValues)