Python - 如何在Python3中获取列表中列表的第一个项目

Question

我有一个Python列表，内容如下：

edge =  [[1, 2], [1, 3], [1, 4], [3, 4]]

我想要打印1和1和1和3; 即每个子列表的第一个元素。

我使用了这段代码：

for subList in edge:
    for firstItem in subList:
        print(firstItem)

但它会打印出所有元素..

- Saeed Rahmani

---

使用列表推导式打印所有第一个元素，print([x[0] for x in edge])。 - balaji k

Answer 1

您正在遍历所有嵌套列表的元素。如果您只想打印每个嵌套列表的第一个元素，请使用索引：

for sublist in edge:
    print(sublist[0])

如果所有嵌套的列表都具有相同数量的元素，您可以使用解包：

for start, end in edge:
    print(start)

- Martijn Pieters

Answer 2

def first_item_generator(edge):
    for sublist in edge:
        yield sublist[0]

edge = [[1, 2], [1, 3], [1, 4], [3, 4]]
generator = first_item_generator(edge)
first_items = list(generator)

print(first_items)

该函数生成并返回每个子列表的第一个元素。然后，我使用该函数创建了一个生成器对象，并使用list(generator)将其转换为列表，以获得最终的第一个元素列表。

- Subham