使用python中的sqlalchemy和pandas构造具有可变长度列表的SQL查询

Question

我想用对任意长度的列表起作用的IN运算符构造一个SQL语句。 我正在使用python，pandas和sqlalchemy。

例如，如果我要执行的查询是

SELECT * FROM users WHERE age IN (25, 26, 27)"

我努力了：

import pandas as pd
from sqlalchemy.sql import text

ages = (25, 26, 27)
query = text("SELECT * FROM users WHERE age IN (:x)")
df = pd.read_sql_query(query, conn, params={"x":ages})

但这会导致错误。 如何正确构造查询？ 我希望该解决方案在IN运算符列表中只有一个值的情况下起作用，例如ages = tuple(25) 。

Answer 1

一种解决方案是使用python的字符串join()方法。

该示例将显示为：

import pandas as pd
from sqlalchemy.sql import text

ages = (25, 26, 27)
ages = (str(n) for n in ages)
ages = ','.join(ages)
query = text("SELECT * FROM users WHERE age IN (:x)")
df = pd.read_sql_query(query, conn, params={"x":ages})

Answer 2

考虑一个用于占位符的动态SQL字符串构建，以及一个对参数使用字典理解的动态字典构建。 下面假设您的RDMS是带有冒号命名参数的SQLite：

import pandas as pd
from sqlalchemy.sql import text

ages = (25, 26, 27)
placeholders = ', '.join([':param'+str(i) for i in range(len(ages))])
paramdict = {'param'+str(i): ages[i] for i in range(len(ages))}

query = text("SELECT * FROM users WHERE age IN ({})".format(placeholders))
# SELECT * FROM users WHERE age IN (:param0, :param1, :param2)

df = pd.read_sql_query(query, conn, params=paramdict)