在 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]]

但我认为可能有更好的语法来实现它。你知道更好的方法吗？

360604 次浏览
小开
最佳答案

使用 列表内涵:

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]
>>>

这两种方法实际上做同样的事情，所以您可以选择您喜欢的任何方法。

小开

你可以使用列表内涵:

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。

小开

如果你因为某些原因不想使用列表内涵，你可以使用 地图和 操作员 itemgetter:

>>> from operator import itemgetter
>>> rows = [(1, 2), (3, 4), (5, 6)]
>>> map(itemgetter(1), rows)
[2, 4, 6]
>>>
小开

实现这一点的功能性方法是使用以下方法解压缩列表:

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,9,26,9,44)


提交答案