如何从Python中的SQL查询中获取单个结果？

Question

使用 Python 时，是否有一种优雅的方式从 SQLite SELECT 查询获取单个结果？

例如：

conn = sqlite3.connect('db_path.db')
cursor=conn.cursor()
cursor.execute("SELECT MAX(value) FROM table")

for row in cursor:
    for elem in row:
        maxVal = elem

有没有办法避免那些嵌套的 for 并直接获取值？我尝试过

maxVal = cursor[0][0]

但没有成功。

Answer 1

我认为您正在寻找 Cursor.fetchone() ：

cursor.fetchone()[0]

Answer 2

请小心，接受的答案可能会导致 TypeError！

由于 fetchone() 文档：

获取查询结果集的下一行，返回单个序列，如果没有更多数据可用，则返回 None。

对于某些 SQL 查询 cursor.fetchone()[0] 可能会变成 None[0] ，从而导致引发 TypeError 异常。

获取第一行或None的更好方法是：

first_row = next(cursor, [None])[0]

如果 SQL 查询为空， next 将使用默认值 [None] 并从该列表中获取第一个元素，而不会引发异常。

Answer 3

或者，您可以编写一个包装函数，在给定 SQL 的情况下，返回一个标量结果：

def get_scalar_result(conn, sql):
    cursor=conn.cursor()
    cursor.execute(sql)

    return cursor.fetchone()[0]

对于上面的 Python 语法可能不太正确，我深表歉意，但我希望您明白这一点。

Answer 4

如果您没有使用内置的 cursor.fetchone pysqlite

cursor.execute("select value from table order by value desc limit 1")

Answer 5

序列解包可用于提取标量值来自结果元组。

如果有多行，则通过迭代游标（或cursor.fetchall）：

for result, in cursor:
    print(result)

或者如果结果集中只有一行，则使用cursor.fetchone：

result, = cur.fetchone()
print(result)

在这两种情况下result 之后的尾随逗号将元素从单元素元组中解压出来。这与更常见的情况相同，

a, b = (1, 2)

只是元组只有一个元素：

a, = (1,)

Answer 6

select count(*) from ... groupy by ... 返回 None 而不是 0，
因此 fetchone()[0] 会导致异常。

所以

def get_scalar_from_sql(sqlcur, sqlcmd):
    # select count(*) from .... groupy by ... returns None instead of 0
    sqlcur.execute(sqlcmd)
    scalar = 0
    tuple_or_None = sqlcur.fetchone()
    if not tuple_or_None is None:
        (scalar,) = tuple_or_None
    return scalar

Answer 7

或者你可以尝试：
cursor.execute("SELECT * FROM table where name='martin'")