Python操作数据库写入CSV文件

这篇博客记录了作者使用Python3.6从数据库获取瓶码数据,并将其写入CSV文件的过程。通过pymysql和csv库实现,内容包括代码解析及如何读取指定列。注意每次执行会覆盖原有CSV数据。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一、背景
在做APP自动化测试数据进行分离的时候,需要将数据库的瓶码数据获取到,然后再传入接口,我在网上边学边做花了两天的时间终于,弄成了理想的样子。在这里记录自己的学习过程。
二、环境
Python3.6
MySQL-Navicat(验证SQL语言是否写正确)
用到的库:
import pymysql
import csv
三、代码解析

# -*- coding: utf-8 -*-
import Common.configDB #这里我对数据库操作进行了封装
import csv

DB = Common.configDB.MyDB()
#定义文件路径
file = "E:\python\Factory_AppTest1\Factory_AppTest\code.csv"

with open(file, 'w+', newline='') as csv_file:#newline=''是为了避免出现空格
    write = csv.writer(csv_file, dial
### 如何用Python将列表数据写入CSV文件Python中,可以使用内置的`csv`模块来处理CSV文件的读写操作。以下是一个完整的解决方案,展示如何将列表中的数据写入CSV文件。 ```python import csv # 定义数据 data = [ ["姓名", "年龄", "性别"], ["张三", 18, "男"], ["李四", 20, "女"], ["王五", 22, "男"] ] # 打开文件写入数据 with open("example.csv", "w", encoding="utf-8", newline="") as file: writer = csv.writer(file) writer.writerows(data) print("数据已成功写入CSV文件!") ``` 上述代码首先定义了一个包含多行数据的列表`data`[^2],然后通过`csv.writer`对象将这些数据逐行写入到名为`example.csv`的文件中。`newline=""`参数用于确保在不同操作系统上生成的CSV文件不会出现多余的空行[^4]。 如果需要写入的是字典形式的数据,可以使用`csv.DictWriter`类。以下是一个示例: ```python import csv # 准备数据 data = [ {"姓名": "张三", "年龄": 18, "性别": "男"}, {"姓名": "李四", "年龄": 20, "性别": "女"}, {"姓名": "王五", "年龄": 22, "性别": "男"} ] # 定义CSV文件路径 csv_file_path = 'output.csv' # 打开文件并创建CSV写入器 with open(csv_file_path, 'w', newline='', encoding='utf-8') as csvfile: fieldnames = ['姓名', '年龄', '性别'] writer = csv.DictWriter(csvfile, fieldnames=fieldnames) # 写入表头 writer.writeheader() # 写入数据 writer.writerows(data) print(f"数据已写入 {csv_file_path}") ``` 这段代码展示了如何将字典列表写入CSV文件,其中`DictWriter`会根据指定的字段名自动匹配字典中的键值。 ### 注意事项 - 确保文件编码为`utf-8`以支持中文字符。 - 使用`newline=""`避免在Windows系统下生成多余的空行。 - 如果需要更复杂的操作,例如处理大数据集或与数据库交互,可以考虑使用`pandas`库[^3]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值