Python编程:使用pymssql库实现数据导入导出功能
发布时间:2023-12-11 09:06:53
pymssql是一个用于连接并与Microsoft SQL Server交互的Python库。它提供了一种简单且方便的方法来执行SQL查询和操作数据库。
以下是使用pymssql库实现数据导入和导出功能的步骤及示例代码:
1. 导入pymssql库:
import pymssql
2. 连接到数据库:
conn = pymssql.connect(server='server_name', user='username', password='password', database='database_name')
替换'server_name'、'username'、'password'和'database_name'为实际的数据库服务器名称、用户名、密码和数据库名称。
3. 创建游标对象:
cursor = conn.cursor()
4. 执行查询语句:
cursor.execute('SELECT * FROM table_name')
替换'table_name'为实际的表名称。
5. 获取查询结果:
result = cursor.fetchall()
可以使用fetchall()方法获取查询结果,也可以使用fetchone()方法获取单个查询结果。
6. 关闭游标和数据库连接:
cursor.close() conn.close()
7. 将查询结果导出为CSV文件:
import csv
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(result)
这将创建一个名为'output.csv'的文件,并将查询结果写入该文件中。
8. 从CSV文件导入数据:
with open('input.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
cursor.execute('INSERT INTO table_name VALUES (%s, %s, %s)', row)
这将打开一个名为'input.csv'的文件,并逐行读取文件中的数据。然后,使用execute()方法将每一行数据插入到数据库表中。
这就是使用pymssql库实现数据导入和导出功能的步骤及示例代码。该库提供了一种简单且灵活的方法来处理与Microsoft SQL Server的交互,可以轻松地执行SQL查询,导出查询结果,并从CSV文件中导入数据。
