linux上使用python pandas操作sqlserver读取写入数据

背景

需求是需要在linux(debian)里运行python程序操作sqlserver读写数据，涉及到数据的读取和写入，这里主要使用pandas进行数据处理。

前提

在运行python程序之前，需要在linux上安装好python和对应的依赖，可以参考以下文章，虽然参考文章都是针对pyodbc连接sqlserver的，但是一开始使用pyodbc的时候，我从数据库读取到的数据一直都是乱码，当我改成pymssql连接的时候就没有乱码了，所以代码中主要是使用pymssql去连接sqlserver。

• Linux下python连接sqlserver
• Linux下Python通过pyodbc连接SQLServer
• linux(debian)系统django配远程连接sqlserver数据库

下面列出一些我参考上面文章用到过的部分命令：

# 解决linux上pip安装pyodbc报错
#src/pyodbc.h:56:10: fatal error: sql.h: No such file or directory
#linux须要安装sqlsrv扩展来连接数据库：
apt-get install unixodbc-dev
apt-get install sqlsrv

# 解决安装tdsodbc报错：debconf: falling back to frontend: Readline：
apt-get -y update && DEBIAN_FRONTEND=noninteractive apt-get -y freetds-dev tdsodbc

#安装pymssql
pip install pymssql

下面列出我安装完依赖服务器上包含的所有包：

# pip list
Package         Version
--------------- -------
greenlet        1.0.0
numpy           1.20.2
pandas          1.2.3
pip             21.0.1
pymssql         2.1.5
pyodbc          4.0.30
python-dateutil 2.8.1
pytz            2021.1
scipy           1.6.2
setuptools      54.2.0
six             1.15.0
SQLAlchemy      1.4.3
wheel           0.36.2

实现

编写python文件

# -*-coding:utf-8-*-
import pandas as pd
import numpy as np
import sqlalchemy
import pymssql
from sqlalchemy import create_engine
import json

# 获取数据库连接
engine = create_engine('mssql+pymssql://username:password@192.168.1.1:1433/database?charset=utf8')

# 创建示例数据集
data={"one":np.random.randn(4),"two":np.linspace(1,4,4),"three":['zhangsan','历史',999,0.1]}
df_write=pd.DataFrame(data,index=[1,2,3,4])

# 写入数据库
df_write.to_sql('test', engine, index=False,if_exists='replace',dtype = {
    'one':  sqlalchemy.types.Float(precision=3, asdecimal=True),
    'two': sqlalchemy.types.Float(precision=3, asdecimal=True),
    'three': sqlalchemy.types.NVARCHAR(length=255)
})

# 查询语句
sql_query_test = '''select * from test;'''

# 读取到DataFrame
df_test = pd.read_sql_query(sql_query_test, engine)

# 打印读取结果
print(df_test)

运行python文件

python demo.py

查看执行结果

控制台：

数据库：

---

参考

解决使用Pyodbc向MSSQL插入unicode字符乱码的问题

通过pymssql驱动程序支持Microsoft SQL Server数据库