python - 如何在 psycopg2 中返回列表而不是元组

Question

我有以下查询

    cursor.execute(
    """
    SELECT transform(row_to_json(t)) FROM 
        (select * from table 
         where a = %s 
         and b = %s limit 1000) t;
    """
    , (a_value, b_value))

运行 records = cursor.fetchall() 将返回大小为 1 的元组列表。

有没有办法只返回一个列表列表？

我问这个是因为我想将列表列表转换为 numpy 矩阵，并且通过循环将单例元组转换为列表很慢。

Answer 1

当你有多于一行时，你可以使用下面的代码

result = [r[0] for r in cur.fetchall()]