在Python中获取列表中每个元组的第一个元素
一个SQL查询给了我一个元组列表,像这样:
[(elt1, elt2), (elt1, elt2), (elt1, elt2), (elt1, elt2), (elt1, elt2), ...]
我想拥有每个元组的所有第一个元素。 现在我用这个:
rows = cur.fetchall() res_list = [] for row in rows: res_list += [row[0]]
但是我认为可能有更好的语法来做到这一点。 你知道更好的方法吗?
使用列表理解 :
res_list = [x[0] for x in rows]
下面是一个演示:
>>> rows = [(1, 2), (3, 4), (5, 6)] >>> [x[0] for x in rows] [1, 3, 5] >>>
或者,您可以使用解包而不是x[0]
:
res_list = [x for x,_ in rows]
下面是一个演示:
>>> lst = [(1, 2), (3, 4), (5, 6)] >>> [x for x,_ in lst] [1, 3, 5] >>>
两种方法几乎都做同样的事情,所以你可以select任何你喜欢的。
如果您不想使用列表理解,可以使用map和operator.itemgetter :
>>> from operator import itemgetter >>> rows = [(1, 2), (3, 4), (5, 6)] >>> map(itemgetter(1), rows) [2, 4, 6] >>>
你可以使用列表理解:
res_list = [i[0] for i in rows]
这应该是诀窍
res_list = [x[0] for x in rows]
参考http://docs.python.org/3/tutorial/datastructures.html#list-comprehensions
有关为什么喜欢理解比map
等更高阶函数的讨论,请访问http://www.artima.com/weblogs/viewpost.jsp?thread=98196 。
实现这个function的方法是使用以下命令解压缩列表:
sample = [(2, 9), (2, 9), (8, 9), (10, 9), (23, 26), (1, 9), (43, 44)] first,snd = zip(*sample) print first,snd (2, 2, 8, 10, 23, 1, 43) (9, 9, 9, 9, 26, 9, 44)