我有一个python列表,说我
l = [1,5,8]
我想编写一个SQL查询来获取列表中所有元素的数据,例如
select name from students where id = |IN THE LIST l|
我该如何完成?
我有一个python列表,说我
l = [1,5,8]
我想编写一个SQL查询来获取列表中所有元素的数据,例如
select name from students where id = |IN THE LIST l|
我该如何完成?
Answers:
到目前为止,答案一直是将这些值模板化为纯SQL字符串。这对于整数绝对没问题,但是如果我们想对字符串进行处理,则会遇到转义问题。
这是一个使用参数化查询的变体,它对两个都适用:
placeholder= '?' # For SQLite. See DBAPI paramstyle.
placeholders= ', '.join(placeholder for unused in l)
query= 'SELECT name FROM students WHERE id IN (%s)' % placeholders
cursor.execute(query, l)
([placeholder]*len(l))
由于占位符可能是多个字符,因此通常情况下必须如此。
cursor.execute(f'SELECT name FROM students WHERE id IN ({','.join('?' for _ in l)})', l)
。@bobince,您还应该在解决方案中指出?
,就避免SQL注入而言,使用是安全的方法。这里有很多容易受到攻击的答案,基本上都是将python中的字符串连接起来的答案。
最简单的方法是将列表转到tuple
第一
t = tuple(l)
query = "select name from studens where id IN {}".format(t)
不要使其复杂化,解决方案很简单。
l = [1,5,8]
l = tuple(l)
params = {'l': l}
cursor.execute('SELECT * FROM table where id in %(l)s',params)
我希望这可以帮助!!!
l
仅包含一个元素,则此操作将无效,最后得到id IN (1,)
。这是语法错误。
sqlite3
。您针对哪个库进行了测试?
您想要的SQL是
select name from studens where id in (1, 5, 8)
如果您想从python构造它,可以使用
l = [1, 5, 8]
sql_query = 'select name from studens where id in (' + ','.join(map(str, l)) + ')'
的地图功能将改变列表转换成可以通过使用逗号胶合在一起的字符串列表str.join方法。
或者:
l = [1, 5, 8]
sql_query = 'select name from studens where id in (' + ','.join((str(n) for n in l)) + ')'
如果您更喜欢生成器表达式而不是map函数。
更新:S. Lott在评论中提到Python SQLite绑定不支持序列。在这种情况下,您可能想要
select name from studens where id = 1 or id = 5 or id = 8
产生者
sql_query = 'select name from studens where ' + ' or '.join(('id = ' + str(n) for n in l))
string.join用逗号分隔的列表值,并使用format运算符形成查询字符串。
myquery = "select name from studens where id in (%s)" % ",".join(map(str,mylist))
(谢谢,布莱尔康拉德)
%
这里使用的存在,只是普通的Python字符串格式化。
我喜欢bobince的回答:
placeholder= '?' # For SQLite. See DBAPI paramstyle.
placeholders= ', '.join(placeholder for unused in l)
query= 'SELECT name FROM students WHERE id IN (%s)' % placeholders
cursor.execute(query, l)
但是我注意到了这一点:
placeholders= ', '.join(placeholder for unused in l)
可以替换为:
placeholders= ', '.join(placeholder*len(l))
如果不太聪明和不太笼统,我会觉得这更直接。这里l
需要有一个长度(即,引用一个定义__len__
方法的对象),这应该不是问题。但是占位符也必须是单个字符。要支持多字符占位符使用:
placeholders= ', '.join([placeholder]*len(l))
@umount答案的解决方案,因为它用一个元素的元组中断,因为(1,)不是有效的SQL。
>>> random_ids = [1234,123,54,56,57,58,78,91]
>>> cursor.execute("create table test (id)")
>>> for item in random_ids:
cursor.execute("insert into test values (%d)" % item)
>>> sublist = [56,57,58]
>>> cursor.execute("select id from test where id in %s" % str(tuple(sublist)).replace(',)',')'))
>>> a = cursor.fetchall()
>>> a
[(56,), (57,), (58,)]
sql字符串的其他解决方案:
cursor.execute("select id from test where id in (%s)" % ('"'+'", "'.join(l)+'"'))
placeholders= ', '.join("'{"+str(i)+"}'" for i in range(len(l)))
query="select name from students where id (%s)"%placeholders
query=query.format(*l)
cursor.execute(query)
这应该可以解决您的问题。
如果您将PostgreSQL与Psycopg2库一起使用,则可以让其元组适应为您完成所有转义和字符串插值,例如:
ids = [1,2,3]
cur.execute(
"SELECT * FROM foo WHERE id IN %s",
[tuple(ids)])
即只需确保您将IN
参数作为传递tuple
。如果是,则list
可以使用= ANY
数组语法:
cur.execute(
"SELECT * FROM foo WHERE id = ANY (%s)",
[list(ids)])
请注意,这两个都将变成相同的查询计划,因此您应该只使用较容易的那个。例如,如果您的列表位于一个元组中,则使用前者;如果它们存储在列表中,则使用后者。
一个更简单的解决方案:
lst = [1,2,3,a,b,c]
query = f"""SELECT * FROM table WHERE IN {str(lst)[1:-1}"""
','.join(placeholder * len(l))
在仍然可读的恕我直言的情况下会短一些