Question

通常，如果我想在表中插入值，我会做这样的事情（假设我知道我要插入的值属于哪些列）：

conn = sqlite3.connect('mydatabase.db')
conn.execute("INSERT INTO MYTABLE (ID,COLUMN1,COLUMN2)\
VALUES(?,?,?)",[myid,value1,value2])

但是现在我有一个列列表（列表的长度可能会有所不同）以及列表中每列的值列表。

例如，如果我有一个包含10列的表（即，column1，column2 ...，column10等）。我有一个我想要更新的列列表。让我们说[column3,column4]。我有这些列的值列表。 [value for column3,value for column4]。

如何将列表中的值插入每个属于的列？

Answer 1

据我所知conn.execute中的参数列表仅适用于值，因此我们必须使用这样的字符串格式：

import sqlite3
conn = sqlite3.connect(':memory:')
conn.execute('CREATE TABLE t (a integer, b integer, c integer)')
col_names = ['a', 'b', 'c']
values = [0, 1, 2]
conn.execute('INSERT INTO t (%s, %s, %s) values(?,?,?)'%tuple(col_names), values)

请注意这是一次非常糟糕的尝试，因为应始终检查传递给数据库的字符串是否存在注入攻击。但是，您可以在插入之前将列名列表传递给某些注入函数。

<强>编辑：
对于具有不同长度的变量，您可以尝试类似

的变量

exec_text = 'INSERT INTO t (' + ','.join(col_names) +') values(' + ','.join(['?'] * len(values)) + ')'
conn.exec(exec_text, values)
# as long as len(col_names) == len(values)

Answer 2

当然字符串格式化会起作用，你只需要对它有点了解。

col_names = ','.join(col_list)
col_spaces = ','.join(['?'] * len(col_list))
sql = 'INSERT INTO t (%s) values(%s)' % (col_list, col_spaces)
conn.execute(sql, values)

Answer 3

我正在寻找一种基于未知/可变长度列表创建列的解决方案，并发现了这个问题。但是，我设法找到了一个更好的解决方案（无论如何对我来说），它也变得更现代了，所以我认为我会包括它，以防它对某人有所帮助：

ValueError: Columns must be same length as key

我所有的数据都是文本类型的，所以我只有一个变量“ col_type”-但是例如，您可以输入一个元组列表（如果您要输入的话，也可以输入一个元组）：

import sqlite3

def create_sql_db(my_list):

    file = 'my_sql.db'
    table_name = 'table_1'
    init_col = 'id'
    col_type = 'TEXT'

    conn = sqlite3.connect(file)
    c = conn.cursor()

    # CREATE TABLE (IF IT DOESN'T ALREADY EXIST)
    c.execute('CREATE TABLE IF NOT EXISTS {tn} ({nf} {ft})'.format(
        tn=table_name, nf=init_col, ft=col_type))

    # CREATE A COLUMN FOR EACH ITEM IN THE LIST
    for new_column in my_list:
        c.execute('ALTER TABLE {tn} ADD COLUMN "{cn}" {ct}'.format(
            tn=table_name, cn=new_column, ct=col_type))

    conn.close()


my_list = ["Col1", "Col2", "Col3"]

create_sql_db(my_list)

并将“创建列”步骤更改为：

my_other_list = [("ColA", "TEXT"), ("ColB", "INTEGER"), ("ColC", "BLOB")]

Python Sqlite3插入操作与列名列表

3 个答案: