Python：我怎樣才能擺脫每個子列表的第二個元素？

Question

我有一個子列表列表，例如：

[[501,4]，[501,4]，[501,4]，[501,4]]

如何擺脫每個子列表的第二個元素？ （即4）

[501,501,501,501]

我應該迭代列表還是有更快的方法？ 謝謝

Answer 1

您可以使用列表推導來獲取每個子列表的第一個元素：

xs = [[501, 4], [501, 4], [501, 4], [501, 4]]
[x[0] for x in xs]
# [501, 501, 501, 501]

Answer 2

a = [[501, 4], [501, 4], [501, 4], [501, 4]]
b = [c[0] for c in a]

Answer 3

使用map pythonic功能較少的版本：

a = [[501, 4], [501, 4], [501, 4], [501, 4]]
map(lambda x: x[0], a)

較少pythonic，因為它不使用列表推導。 看到這里 。