pymssql是一个用于连接Microsoft SQL Server数据库的Python库。它提供了一组函数和方法,可以方便地进行数据库操作。下面是使用pymssql写入CSV并使用列名作为标题的步骤:
import pymssql
import csv
conn = pymssql.connect(server='服务器地址', user='用户名', password='密码', database='数据库名')
cursor = conn.cursor()
请将服务器地址
、用户名
、密码
、数据库名
替换为实际的数据库连接信息。
cursor.execute('SELECT 列名1, 列名2, 列名3 FROM 表名')
rows = cursor.fetchall()
请将列名1
、列名2
、列名3
替换为实际的列名,将表名
替换为实际的表名。
with open('output.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerow([column[0] for column in cursor.description]) # 写入列名作为标题
writer.writerows(rows) # 写入查询结果数据
这里创建了一个名为output.csv
的CSV文件,并将列名作为标题写入文件的第一行,然后将查询结果数据写入文件。
cursor.close()
conn.close()
完整的代码示例如下:
import pymssql
import csv
conn = pymssql.connect(server='服务器地址', user='用户名', password='密码', database='数据库名')
cursor = conn.cursor()
cursor.execute('SELECT 列名1, 列名2, 列名3 FROM 表名')
rows = cursor.fetchall()
with open('output.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerow([column[0] for column in cursor.description]) # 写入列名作为标题
writer.writerows(rows) # 写入查询结果数据
cursor.close()
conn.close()
这样,你就可以使用pymssql库将SQL查询结果写入CSV文件,并使用列名作为标题。注意,这只是一个简单的示例,实际应用中可能需要根据具体需求进行适当的修改和扩展。
领取专属 10元无门槛券
手把手带您无忧上云