如何在Python中保存数据到CSV文件中

分类:知识百科 日期: 点击:0

在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()函数关闭文件对象,以完成文件的保存。

标签:

版权声明

1. 本站所有素材,仅限学习交流,仅展示部分内容,如需查看完整内容,请下载原文件。
2. 会员在本站下载的所有素材,只拥有使用权,著作权归原作者所有。
3. 所有素材,未经合法授权,请勿用于商业用途,会员不得以任何形式发布、传播、复制、转售该素材,否则一律封号处理。
4. 如果素材损害你的权益请联系客服QQ:77594475 处理。