在Python中保存数据到CSV文件中,可以使用csv模块中的writer()函数来实现。writer()函数可以接受一个文件对象,并将行写入文件中。
示例代码:
import csv # 创建一个文件对象 with open('data.csv', 'w', newline='') as csvfile: # 创建一个writer对象 writer = csv.writer(csvfile) # 写入行数据 writer.writerow(['id', 'name', 'age']) writer.writerow(['1', 'John', '20']) writer.writerow(['2', 'Smith', '30']) writer.writerow(['3', 'Amy', '25'])
上面的代码中,使用open()函数创建一个文件对象,并将其传递给csv.writer()函数,以创建一个writer对象。使用writer.writerow()函数写入行数据。
还可以使用csv.DictWriter类来写入字典数据。它可以接受一个字典对象,并将其写入文件中。
示例代码:
import csv # 创建一个文件对象 with open('data.csv', 'w', newline='') as csvfile: # 创建一个writer对象 fieldnames = ['id', 'name', 'age'] writer = csv.DictWriter(csvfile, fieldnames=fieldnames) # 写入行数据 writer.writeheader() writer.writerow({'id': '1', 'name': 'John', 'age': '20'}) writer.writerow({'id': '2', 'name': 'Smith', 'age': '30'}) writer.writerow({'id': '3', 'name': 'Amy', 'age': '25'})
上面的代码中,使用open()函数创建一个文件对象,并将其传递给csv.DictWriter()函数,以创建一个writer对象。使用writer.writeheader()函数写入表头,再使用writer.writerow()函数写入行数据。
使用close()函数关闭文件对象,以完成文件的保存。