pd.read_sql_query字符串列表参数折叠到第一个元素

时间:2020-08-25 17:38:10

标签: pandas python-3.7

我在使用pandas.read_sql_query()方法时遇到了一个奇怪的问题,当我传递参数(该参数是格式为字符串的列表)时,该参数仅读取列表的第一个元素,并删除其他所有内容。原始列表是将数字重新格式化为字符串的列表,因此,从理论上讲,应保留列表的所有元素。例如,此代码返回的数据框:

    l = [1, 2, 3]
    l_as_str =  ", ".join([x for x in map(str, l)])
    
    QUERY = """SELECT * FROM table WHERE id in (%(l_as_str)s)"""
    
    df = pd.read_sql_query(QUERY, params={"l_as_str": l_as_str}, con=engine) 

...仅包含id = 1的项目,l的第一个元素。如果我切换元素的顺序(例如l = [2, 1, 3]),则它仅返回id = 2的项目。换句话说,l_as_str似乎只折叠到第一个元素。知道是什么问题吗?

1 个答案:

答案 0 :(得分:0)

只需使用f字符串简化即可。无需通过params

l = [1, 2, 3]
l_as_str =  ", ".join([x for x in map(str, l)])
    
QUERY = f"""SELECT * FROM table WHERE id in ({l_as_str})"""
df = pd.read_sql_query(QUERY, con=engine)