python连接sql server数据库

Posted 2021-09-04 AI算法攻城狮

1、读取数数据保存到csv

import pymssql #引入pymssql模块
import pandas as pd

def conn():
    connect = pymssql.connect('', '', '', '') #服务器名,账户,密码,数据库名
    if connect:
        print("连接成功!")
    return connect


if __name__ == '__main__':
    conn = conn()
    s="select * from XXX"
    data = pd.read_sql(s, conn)
    if len(data) > 1:
        path = "cost.csv"
        xlsxdata = pd.DataFrame(data=data)
        xlsxdata.to_csv(path, index=False)
    conn.close()

2、遍历数据

import pymssql #引入pymssql模块

def conn():
    connect = pymssql.connect('', '', '', '') #服务器名,账户,密码,数据库名
    if connect:
        print("连接成功!")
    return connect


if __name__ == '__main__':
    conn = conn()
    cursor = conn.cursor()
    cursor.execute('select * from XXX')
    # 获取一条记录
    row = cursor.fetchone()
    # 循环打印记录(这里只有一条，所以只打印出一条)
    while row:
        print("dt=%d, Name=%s" % (row[0], row[1]))
        row = cursor.fetchone()
    # 连接用完后记得关闭以释放资源
    conn.close()