Python连接SQL Server

使用Python连接SQL Server进行取数操作
输入：SQL
输出：DataFrame

# 导入连接sql server的包
import pymssql

def sql_data(SQL):
	# 设定变量连接数据库
    conn = pymssql.connect(
        database = 'database',
        user = 'user',
        password = 'password',
        server = 'server ip'
    ) 
    # 创建光标
    cursor = conn.cursor()
    # 执行SQL语句
    cursor.execute(SQL)
    # 获取结果
    results = cursor.fetchall()
    # 获取数据的表头
    data_dict=[]
    for field in cursor.description:
        data_dict.append(field[0])
    
    # 把数据放进DataFrame
    data = pd.DataFrame(results, columns=data_dict)
    # 关闭光标
    conn.close
    # 关闭连接
    cursor.close
    
    return data


# 写入SQL命令
SQL = ''' 
SELECT * FROM [table]
'''
# 获取数据
df = sql_data(SQL)
df