python - 我可以将 Python Pandas 数据框导出到 MS SQL 吗？

Question

我正在使用 pymssql 和 Pandas sql 包将数据从 SQL 加载到带有 frame_query 的 Pandas dataframe 中。

我想使用 write_frame 将其发送回 SQL 数据库，但我找不到太多关于此的文档。特别是有一个参数flavor='sqlite'。这是否意味着到目前为止 Pandas 只能导出到 SQLite？我的公司使用的是 MS SQL Server 2008，所以我需要导出到它。

Answer 1

不幸的是，是的。目前 sqlite 是 write_frame 支持的唯一“风格”。参见 https://github.com/pydata/pandas/blob/master/pandas/io/sql.py#L155

def write_frame(frame, name=None, con=None, flavor='sqlite'):
    """
    Write records stored in a DataFrame to SQLite. The index will currently be
    dropped
    """
    if flavor == 'sqlite':
        schema = get_sqlite_schema(frame, name)
    else:
        raise NotImplementedError

不过，编写一个简单的 write_frame 应该相当容易。例如，这样的事情可能会起作用(未经测试!):

import pymssql                                                        
conn = pymssql.connect(host='SQL01', user='user', password='password', database='mydatabase')
cur = conn.cursor()                                                   

# frame is your dataframe                                             
wildcards = ','.join(['?'] * len(frame.columns))                      
data = [tuple(x) for x in frame.values]
table_name = 'Table' 

cur.executemany("INSERT INTO %s VALUES(%s)" % (table_name, wildcards), data)
conn.commit()