来自psycopg2 PostgreSQL查询的Python数组操作

Question

我正在使用psycopg2进行psql查询。

cur.execute("SELECT DISTINCT name FROM product")
result = cur.fetchall()
print(result)

[('product1',), ('product2',), ('product3',), ('product4',)]

我需要重新格式化此数组以创建API端点。 现在，它是一个元组列表，其中元组的第二个值为空。 一个简单的循环遍历就可以完成工作。

results=[]
for item in result:
    results.append(item[0])
print(results)

['product1','product2','product3','product4']

但是，此查询可能会变得很大。 遍历整个列表会增加查询的延迟，这似乎没有必要。 有没有办法在恒定时间内展平数组，或者有其他psycopg2函数以我需要的格式返回？

Answer 1

汇总查询：

query = '''
    select array_agg(distinct name)
    from product
'''
cursor.execute(query)
rs = cursor.fetchall()[0][0]
print rs

输出：

['product1', 'product2', 'product3', 'product4']

Answer 2

与数据库查询相比，转换列表所需的时间可以忽略不计。 但是您无需直接创建列表：

cur.execute("SELECT DISTINCT name FROM product")
result = [item for item, in cur]
print(result)