pandas - 从 DataFrame 插入时获取身份 key

Question

假设我在 SQL Server 中有下表:

CREATE TABLE Users
(
     UserID          int IDENTITY(1,1) PRIMARY KEY
,    UserName        varchar(50)
,    InsertDate      datetime2 DEFAULT GETDATE()
)

在 Python 中，我想从 DataFrame 插入到这个表中并取回 UserID我刚刚插入的行

import numpy as np
import pandas as pd
import sqlalchemy as sa

engine = sa.create_engine(...)
df = pd.DataFrame({'UserName': ['Alice', 'Bob']})
df.to_sql('Users', engine, if_exists='append', index=False)

数据正在进入数据库。但是我怎样才能找回 UserID我刚刚插入的行？我考虑编写一个包装函数，首先插入临时表并使用 T-SQL 操作数据并返回我想要的内容。但是有没有更简单的方法呢？

备注 :无论如何我都无法修改表的架构。我只能在数据库中创建临时表。

Answer 1

要取回数据......你只需要查询表。

with engine.connect() as conn:
    df=pd.read_sql("select * from Users",conn)