游标执行后返回的结果都只是数据,但是不带有列名标识。这里需要处理2个问题:
- 将返回的数据映射到每一列上
- 当返回的结果很大的时候,需要使用迭代器来提升性能。
解决上面的2个问题,在python里面可以采用下面的2种方式来处理。
- 使用namedtuple 和 map object。
- 使用yield 和 zip。
下面是示例代码:
![](https://www.cnblogs.com/Images/OutliningIndicators/ContractedBlock.gif)
![](https://www.cnblogs.com/Images/OutliningIndicators/ExpandedBlockStart.gif)
Result_From_DB# MySQL 数据库 import mysql from mysql import connector from collections import namedtuple def generate_namedtuple(cur): from collections import namedtuple fieldnames = [d[0].lower() for d in cur.description] Record = namedtuple('Record', fieldnames) rows = cur.fetchall() if not rows:return else: return map(Record._make, rows) def generate_dicts(cur): fieldnames = [d[0].lower() for d in cur.description] while True: rows = cur.fetchmany() if not rows: return for row in rows: yield dict(zip(fieldnames, row)) if __name__ == '__main__': user = 'herbert' pwd = '851020' host = '127.0.0.1' db = 'world' cnx = mysql.connector.connect(user=user, password=pwd, host=host,database=db) cur = cnx.cursor() cur.execute("SELECT Name, CountryCode, District, Population FROM CITY\ where CountryCode = 'CHN' AND Population > 500000") for r in generate_dicts(cur): print(r['name'], r['population']) cur.execute("SELECT Name, CountryCode, District, Population FROM CITY\ where CountryCode = 'CHN' AND Population > 500000") print("-----------------------------") for k in generate_namedtuple(cur): print(k.name, k.population) cur.close() cnx.close()
需要注意几点:
- 使用map和namedtuple的时候,游标要使用fetchall()方法。一次行取出所有结果,然后调用map方法将所有的数据map到Record object上。
- 返回的map object可以调用for方法进行遍历。map object类似一个Record object列表。
- namedtuple生成的对象,访问的时候是用dot来访问数据的。
- 使用yield和dict zip返回的是iterator对象,这在性能上应该更有优势。
- cur.fetchmany()返回的rowcount是有cursor.arraysize决定的。默认的是1,可以自行决定每次返回的数量。cur.fetchmany(size)
- 使用yield和dict zip返回的iterator对象,每一个是一个dict对象。