欢迎访问宙启技术站
智能推送

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文件中导入数据。