如何通过python更新sqlite数据库中的特定单元格

Question

我有一个充满数字但作为字符串的数据库。我的主要目标是在 SQLite 数据库中获取一个单元格，将其转换为浮点数，然后用它更新单元格。但是我不明白如何更新 SQLite 数据库中的特定单元格。这是我的代码。

for row in cur:
    for col in row:
        try:
            fcol = float(col)
            cur.execute('UPDATE test SET ')
        except:
            print("FAIL")
            print(col)
            continue
        print("Replaced",col)
    print("Row done")
print("Completed")

基本上我的问题是如何更新 SQLite 数据库的特定单元格。

Answer 1

问题是列被定义为TEXT，因此在获取时值总是字符串。有几种方法可以解决这个问题。

首先，您可以在从数据库中检索它们后将它们转换为float 或Decimal。这既不方便又容易出错。

更好的方法可能是创建一个新列，类型为REAL，并使用旧列中的数据加载它。此示例创建并填充一个表，然后演示如何添加一个新列。

import random
import sqlite3


with sqlite3.connect('test.db') as conn:
    cur = conn.cursor()
    cur.execute('DROP TABLE IF EXISTS test')
    cur.execute('CREATE TABLE test (id INTEGER, col TEXT)')
    cur.executemany('INSERT INTO test (id, col) VALUES (?, ?)',
                    [(i, random.random(),) for i in range(3)])
    conn.commit()
    cur.execute('SELECT col FROM test')
    for row in cur:
        print(row)
    print()
    cur.execute('ALTER TABLE test ADD COLUMN new_col REAL')
    cur.execute('UPDATE test SET new_col = CAST(col AS REAL)')
    conn.commit()
    try:
        # Not all versions of Sqlite support DROP COLUMN
        cur.execute('ALTER TABLE test DROP COLUMN col')
        conn.commit()
    except sqlite3.OperationalError:
        conn.rollback()
    cur.execute('SELECT id, new_col FROM test')
    for row in cur:
        print(row)
    print()

第三，您可以创建一个具有正确数据类型的新表并从原始表中复制数据。和之前一样，这个示例创建并填充了一个表，然后演示了如何复制到一个新表。

with sqlite3.connect('test.db') as conn:
    cur = conn.cursor()
    cur.execute('DROP TABLE IF EXISTS test')
    cur.execute('CREATE TABLE test (id INTEGER, col TEXT)')
    cur.executemany('INSERT INTO test (id, col) VALUES (?, ?)', 
                    [(i, random.random(),) for i in range(3)])
    conn.commit()
    cur.execute('SELECT col FROM test')
    for row in cur:
        print(row)
    print()
    cur.execute('DROP TABLE IF EXISTS test2')
    cur.execute('CREATE TABLE test2 (id INTEGER, col REAL)')
    # The order of SELECTed columns must match the target columns.
    cur.execute("""INSERT INTO test2 (id, col)
                   SELECT t1.id, CAST(t1.col AS REAL) 
                   FROM test t1 
                   WHERE true""")
    conn.commit()
    cur.execute('SELECT id, col FROM test2')
    for row in cur:
        print(row)
    print()

一旦您对结果感到满意，您就可以删除原始表，甚至可以根据需要重命名新表。

在示例中，我每次都删除并重新创建原始表 - 您不想对您的表执行此操作！示例使用简单的两列表，但逻辑可以扩展为处理多列。