如何将元组元组转换为一行列表(pythonic)?

时间:2011-11-17 17:40:55

标签: list cursor tuples python

query = 'select mydata from mytable'
cursor.execute(query)
myoutput = cursor.fetchall() 
print myoutput

(('aa',), ('bb',), ('cc',))

为什么它(cursor.fetchall)返回一个元组元组而不是一个元组,因为我的查询只要求一列数据?

将其转换为['aa', 'bb', 'cc']的最佳方式是什么?

我可以这样做:

mylist = []
myoutput = list(myoutput)
for each in myoutput:
   mylist.append(each[0])

我确信这不是最好的方法。请赐教!

6 个答案:

答案 0 :(得分:7)

这也有效:

>>> tu = (('aa',), ('bb',), ('cc',))
>>> import itertools
>>> list(itertools.chain(*tu))
['aa', 'bb', 'cc']

修改 Could you please comment on the cost tradeoff? (for loop and itertools)

Itertools明显更快:

>>> t = timeit.Timer(stmt="itertools.chain(*(('aa',), ('bb',), ('cc',)))")
>>> print t.timeit()
0.341422080994
>>> t = timeit.Timer(stmt="[a[0] for a in (('aa',), ('bb',), ('cc',))]")
>>> print t.timeit()
0.575773954391

修改2 Could you pl explain itertools.chain(*)

*将序列解压缩为位置参数,在本例中是嵌套的元组元组。

示例:

>>> def f(*args):
...    print "len args:",len(args)
...    for a in args:
...       print a
... 
>>> tu = (('aa',), ('bb',), ('cc',))
>>> f(tu)
len args: 1
(('aa',), ('bb',), ('cc',))
>>> f(*tu)
len args: 3
('aa',)
('bb',)
('cc',)

另一个例子:

>>> f('abcde')
len args: 1
abcde
>>> f(*'abcde')
len args: 5
a
b
c
d
e

请参阅the documents on unpacking

答案 1 :(得分:4)

你可以做到

>>> tup = (('aa',), ('bb',), ('cc',))
>>> lst = [a[0] for a in tup]
>>> lst
['aa', 'bb', 'cc']

答案 2 :(得分:2)

你正在做的是正确但更简洁,可能表现更好

>>> [item for item, in (('aa',), ('bb',), ('cc',)) ]
['aa', 'bb', 'cc']

或者如果您讨厌for关键字,则可以使用地图

>>> map(lambda a:a[0], (('aa',), ('bb',), ('cc',)) )
['aa', 'bb', 'cc']

这是另一种方式

>>> reduce(lambda a, b:a+b, (('aa',), ('bb',), ('cc',)) )
('aa', 'bb', 'cc')

虽然IMO列表理解最具可读性

答案 3 :(得分:2)

  

为什么cursor.fetchall()返回元组元组而不是元组元组,因为我的查询只要求一列数据?

外部元组是完整的结果;每个内部元组代表该结果中的一条记录;因为你只要求一个字段,每个内部元组只有一个元素。

  

将它转换为['aa','bb','cc']的最佳方式是什么?

有几种方法,哪种方式“最好”取决于你在做什么......

简单列表理解:

mylist = [each[0] for each in myoutput]

简单生成器(节省内存使用):

mygen = (each[0] for each in myoutput)
for result in mygen:
    print result

如果您只需要处理myoutput中的项目,也可以执行

for each, in myoutput:
    print each

如果您已经对代码进行了分析并发现这是一个瓶颈,那么您可以提高可读性但速度更快:

import itertools
mylist = list(itertools.chain(*myoutput))

或者,如果您只是需要处理它,那么:

import itertools
for result in itertools.chain(*myoutput):
    print result

答案 4 :(得分:1)

这有效:

>>> tups=(('aa',), ('bb',), ('cc',))
>>> list(*(zip(*tups)))
['aa', 'bb', 'cc']

说明:

1) *tups unpacks the nested tuples  ('aa'),('bb'),('cc')
2) zip produces a list of a single tuple with all the elements: [('aa','bb','cc')]
3) * unpacks that  into  'aa', 'bb', 'cc'
4) creates a list from that unpacking.

你也可以这样做:

>>> list(zip(*tups)[0])
['aa', 'bb', 'cc']

答案 5 :(得分:0)

执行这样的列表理解:

mylist = [ x[0] for x in myoutput ]
相关问题