python - 如何在 Python 中索引嵌套列表？

Question

我有一个嵌套列表，如下所示:

A = [('a', 'b', 'c'),
     ('d', 'e', 'f'),
     ('g', 'h', 'i')]

我正在尝试使用代码打印每个列表的第一个元素:

A = [('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'h', 'i')]
print A[:][0]

但我得到以下输出:

('a', 'b', 'c')

要求的输出:

('a', 'd', 'g')

如何在 Python 中获取此输出？

Answer 1

A[:] 只是创建整个列表的副本，之后您将获得该副本的元素 0。

你需要在这里使用列表理解:

[tup[0] for tup in A]

获取列表，或使用 tuple() 和生成器表达式来获取元组:

tuple(tup[0] for tup in A)

演示:

>>> A = [('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'h', 'i')]
>>> [tup[0] for tup in A]
['a', 'd', 'g']
>>> tuple(tup[0] for tup in A)
('a', 'd', 'g')