在本教程中,你将学习如何用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)