Python连接PostgreSQL查询单列返回扁平列表的方法及原理咨询

为什么返回带逗号的元组结构 #

psycopg2 遵循 Python DB API 2.0 规范要求，fetchall()、fetchone()等查询方法统一返回按行组织的序列结构，每行结果固定用元组表示：

• 多列查询时，元组对应每行的多个字段值，格式为(字段1值, 字段2值, ...)
• 单列查询时，自动生成单元素元组，格式为(字段值,)

单元素元组末尾的逗号是Python语法强制要求：如果没有逗号，(5523)会被Python直接解析为整数5523，无法和多列查询的元组结构保持统一，因此单列查询结果的每行都会显示为带逗号的单元素元组。

方法1：调用时手动处理（无需修改原有函数） #

对db_fetchquery返回的元组列表做列表推导，提取每个元组的第一个元素即可：

# 原有查询逻辑
raw_result = db_fetchquery("SELECT token FROM script")
# 转换为扁平列表
token_list = [row[0] for row in raw_result] if raw_result else []
print(token_list)
# 输出结果：[5523, 5145, 2899]

方法2：修改查询函数，自动适配单列场景 #

如果经常需要查询单列，可以修改db_fetchquery函数的返回逻辑，自动判断列数做压平处理，兼容原有多列查询逻辑：

def db_fetchquery(sql):
    conn = psycopg2.connect(database="trade", user='postgres', password='jps', host='127.0.0.1', port= '5432')
    cursor = conn.cursor()
    conn.autocommit = True
    cursor.execute(sql)
    row = cursor.rowcount
    if row >= 1:
        data = cursor.fetchall()
        conn.close()
        # 新增单列自动压平逻辑
        if len(data[0]) == 1:
            return [row[0] for row in data]
        return data
    conn.close()
    return False

# 直接调用即可返回扁平列表
print(db_fetchquery("SELECT token FROM script"))

内容的提问来源于stack exchange，提问作者bowih