52

我想在列表中获取 fetchall 操作的结果,而不是元组的元组或字典的元组。例如,

cursor = connection.cursor() #Cursor could be a normal cursor or dict cursor
query = "Select id from bs"
cursor.execute(query)
row = cursor.fetchall()

现在,问题是结果行是 ((123,),(234,)) 或 ({'id':123}, {'id':234}) 我正在寻找的是 (123,234) 或 [ 123,234]。如果我可以节省解析结果集,那就最好了。提前致谢

4

7 回答 7

84

那么列表推导呢?如果结果是((123,), (234,), (345,))

>>> row = [item[0] for item in cursor.fetchall()]
>>> row
[123, 234, 345]

如果结果是({'id': 123}, {'id': 234}, {'id': 345})

>>> row = [item['id'] for item in cursor.fetchall()]
>>> row
[123, 234, 345]
于 2012-10-12T21:35:12.037 回答
19

我敢肯定,经过这么长时间,你已经解决了这个问题,但是,对于一些可能不知道如何使用 MySQLdb 将游标的值作为字典获取的人,你可以使用这里找到方法:

import MySQLdb as mdb

con = mdb.connect('localhost', 'testuser', 'test623', 'testdb')

with con:

    cur = con.cursor(mdb.cursors.DictCursor)
    cur.execute("SELECT * FROM Writers LIMIT 4")

    rows = cur.fetchall()

    for row in rows:
        print row["Id"], row["Name"]
于 2013-09-03T18:06:14.203 回答
11

这个旧的 Q 在搜索扁平化数据库查询时出现在 Google 上,所以这里有更多建议......

考虑一个快速列表展平迭代器

其他答案使用fetchall()它首先将所有行加载到内存中,然后对其进行迭代以创建一个新列表。可能效率低下。可以结合 MySQL 所谓的服务器端游标

# assume mysql on localhost with db test and table bs
import itertools
import MySQLdb
import MySQLdb.cursors

conn = MySQLdb.connect(host='localhost',db='test', 
          cursorclass=MySQLdb.cursors.SSCursor ) 
cursor = conn.cursor()
# insert a bunch of rows
cursor.executemany('INSERT INTO bs (id) VALUES (%s)',zip(range(1,10000)) )
conn.commit()
# retrieve and listify
cursor.execute("select id from bs")
list_of_ids = list(itertools.chain.from_iterable(cursor))
len(list_of_ids)
#9999
conn.close()

但是这个问题也被标记为 Django,它有一个很好的单字段查询扁平化器

class Bs(models.Model):
    id_field = models.IntegerField()

list_of_ids = Bs.objects.values_list('id_field', flat=True)
于 2015-04-21T10:22:20.310 回答
4

以这种方式制作光标对象:

db = MySQLdb.connect("IP", "user", "password", "dbname")

cursor = db.cursor(MySQLdb.cursors.DictCursor)

然后,当您对查询执行 cursor.fetchall() 时,将获得一个字典元组,您可以稍后将其转换为列表。

data = cursor.fetchall()

data = list(data)
于 2018-06-20T03:35:03.537 回答
2
list= [list[0] for list in cursor.fetchall()]

这会将结果呈现在一个列表中,例如 - list = [122,45,55,44...]

于 2020-05-20T08:35:51.070 回答
1

如果只有一个字段,我可以使用它从数据库中创建一个列表:

def getFieldAsList():
    kursor.execute("Select id from bs")
    id_data = kursor.fetchall()
    id_list = []
    for index in range(len(id_data)):
        id_list.append(id_data[index][0])
    return id_list
于 2013-02-28T08:25:50.030 回答
-8
cursor.execute("""Select * From bs WHERE (id = %s)""",(id))

cursor.fetchall()
于 2013-05-17T22:55:12.753 回答