我试图在sqlalchemy中执行此查询

SELECT id, name FROM user WHERE id IN (123, 456)

我想在执行时绑定列表[123,456]。


当前回答

或者也可以使用.in_(list),类似于@Carl已经建议的方法 作为

 stmt = select(
        id,
         name
      ).where(
        id.in_(idlist),
      )

完整的代码,假设你在User类中有数据模型:

def fetch_name_ids(engine, idlist):
    # create an empty dataframe
    df = pd.DataFrame()
    try:
        # create session with engine
        session = Session(engine, future=True)
         stmt = select(
            User.id,
            User.name
          ).where(
            User.id.in_(idlist),
          )
        data = session.execute(stmt)

        df = pd.DataFrame(data.all())
        if len(df) > 0:
            df.columns = data.keys()
        else:
            columns = data.keys()
            df = pd.DataFrame(columns=columns)

    except SQLAlchemyError as e:
        error = str(e.__dict__['orig'])
        session.rollback()
        raise error
     else:

        session.commit()
     finally:

        engine.dispose()
        session.close()


      return df

其他回答

只是对上述答案的补充。

如果你想用"IN"语句执行SQL,你可以这样做:

ids_list = [1,2,3]
query = "SELECT id, name FROM user WHERE id IN %s" 
args = [(ids_list,)] # Don't forget the "comma", to force the tuple
conn.execute(query, args)

两个点:

IN语句不需要括号(如“…(% s)),只要写上“…在% s” 强制id列表为元组的一个元素。不要忘记“,”:(ids_list,)

编辑 注意,如果list的长度为1或0,将会引发一个错误!

只是想在python3中使用sqlalchemy和pandas分享我的解决方案。也许,有人会发现它有用。

import sqlalchemy as sa
import pandas as pd
engine = sa.create_engine("postgresql://postgres:my_password@my_host:my_port/my_db")
values = [val1,val2,val3]   
query = sa.text(""" 
                SELECT *
                FROM my_table
                WHERE col1 IN :values; 
""")
query = query.bindparams(values=tuple(values))
df = pd.read_sql(query, engine)

或者也可以使用.in_(list),类似于@Carl已经建议的方法 作为

 stmt = select(
        id,
         name
      ).where(
        id.in_(idlist),
      )

完整的代码,假设你在User类中有数据模型:

def fetch_name_ids(engine, idlist):
    # create an empty dataframe
    df = pd.DataFrame()
    try:
        # create session with engine
        session = Session(engine, future=True)
         stmt = select(
            User.id,
            User.name
          ).where(
            User.id.in_(idlist),
          )
        data = session.execute(stmt)

        df = pd.DataFrame(data.all())
        if len(df) > 0:
            df.columns = data.keys()
        else:
            columns = data.keys()
            df = pd.DataFrame(columns=columns)

    except SQLAlchemyError as e:
        error = str(e.__dict__['orig'])
        session.rollback()
        raise error
     else:

        session.commit()
     finally:

        engine.dispose()
        session.close()


      return df

通过表达式API(基于注释的表达式API是这个问题所要求的),您可以使用相关列的in_方法。

查询

SELECT id, name FROM user WHERE id in (123,456)

use

myList = [123, 456]
select = sqlalchemy.sql.select([user_table.c.id, user_table.c.name], user_table.c.id.in_(myList))
result = conn.execute(select)
for row in result:
    process(row)

这里假设user_table和conn已经适当地定义了。

假设你使用声明式风格(即ORM类),这是相当简单的:

query = db_session.query(User.id, User.name).filter(User.id.in_([123,456]))
results = query.all()

db_session是你的数据库会话,而User是ORM类,__tablename__等于"users"。