Fetchall在Python中只返回一列?

时间:2013-01-07 11:28:03

标签: python mysql-python

我有一个这样的代码:

db = MySQLdb.connect(user='root', db='galaxy', passwd='devil', host='localhost')
cursor = db.cursor()
cursor.execute('SELECT username, password FROM galaxy_user')
names = [row[0] for row in cursor.fetchall()]
passw = [password[1] for password in cursor.fetchall()]
db.close()

问题是我只能从以下代码访问名称或密码。有了这个,我只能得到用户名。我得到了passw的空列表。现在,如果我切换它:

passw = [row[1] for row in cursor.fetchall()]
names = [password[1] for password in cursor.fetchall()

我得到passw的值,但名字是空列表。发生了什么?

1 个答案:

答案 0 :(得分:13)

每次cursor.execute后,您只能使用cursor.fetchall一次。它“耗尽”光标,获取其所有数据,然后无法再次“读取”。

使用以下代码,您可以同时读取所有数据:

db = MySQLdb.connect(user='root', db='galaxy', passwd='devil', host='localhost')
cursor = db.cursor()
cursor.execute('SELECT username, password FROM galaxy_user')
names, passw = zip(*cursor.fetchall())
db.close()

另一种可能性是将所有数据存储在列表中,然后将其读取为光标:

records = cursor.fetchall()
names = [record[0] for record in records]
passw = [record[1] for record in records]

或者字典怎么样(名字 - >密码)?

user_pass = dict(cursor.fetchall())

或简单地(如@JonClemens建议的那样):

user_pass = dict(cursor)