使用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