有时候我们需要一个list列表来查询多条数据,来避免多次执行sql语句,下面就通过一个简单的demo来实现这个功能

  • 连接数据库
import sqlite3

#连接数据库
conn = sqlite3.connect('test.db')
#获取游标用来执行sql语句
c = conn.cursor()
  • 建表
    我们创建一个company表,里面包含了员工的idnameageaddresssalary信息
def create_table():
    c.execute('''CREATE TABLE COMPANY
           (ID INT PRIMARY KEY     NOT NULL,
           NAME           TEXT    NOT NULL,
           AGE            INT     NOT NULL,
           ADDRESS        CHAR(50),
           SALARY         REAL);''')
  • 插入数据
def insert_data_to_table():
    c.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
          VALUES (1, 'Paul', 32, 'California', 10000.00 )")
    c.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
          VALUES (2, 'Allen', 25, 'Texas', 15000.00 )")
    c.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
          VALUES (3, 'Teddy', 23, 'Norway', 20000.00 )")
    c.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
          VALUES (4, 'Mark', 25, 'Rich-Mond ', 65000.00 )")
  • 查询数据
    通过where.......in来实现根据一个list来查询符合条件的多条数据
def select_data(name_list):
    sql = "select * from COMPANY where NAME in ({seq})".format(
        seq=','.join(['?'] * len(name_list)))
    c.execute(sql, name_list)
    return c.fetchall()
#创建commpany表
#create_table()
#插入数据
# insert_data_to_table()
print(select_data(["Paul","Allen"]))
"""
[(1, 'Paul', 32, 'California', 10000.0), (2, 'Allen', 25, 'Texas', 15000.0)]
"""
conn.commit()
conn.close()

参考:
1.https://stackoverflow.com/questions/5766230/select-from-sqlite-table-where-rowid-in-list-using-python-sqlite3-db-api-2-0
2.https://www.runoob.com/sqlite/sqlite-python.html

Logo

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。

更多推荐