python - 从 list of list 的列表中提取第 n 个元素

Question

我有一个列表的列表，我想从每个子列表中提取第 n 个元素。例如，给定以下列表:

my_list = [[[0, 0], [1, 1]], [[2, 0], [3, 2]], [[2, 0], [3, 3]], [[4, 0], [5, 4]], [[6, 0], [7, 5]]]

我想提取所有保持原始列表结构(列表列表列表)的第一个元素，例如:

all_first_elements = [[[0],[1]], [[2],[3]], [[2],[3]], [[4],[5]], [[6],[7]]]

问题类似于this one但有一个额外的嵌套列表。

我试过 all_first_elements = [item[0] for item in my_list] 但它返回列表列表的第一个元素(而不是列表列表的第一个元素)。

Answer 1

您可以在列表解析中使用双 for 循环

all_first_elements = [[[item_1[0]] for item_1 in item_0] for item_0 in my_list]

输出将是

[[[0], [1]], [[2], [3]], [[2], [3]], [[4], [5]], [[6], [7]]]