Skip to main content
 首页 » 编程设计

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

2025年05月04日19zfyouxi

我有一个列表的列表,我想从每个子列表中提取第 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] 但它返回列表列表的第一个元素(而不是列表列表的第一个元素)。

请您参考如下方法:

您可以在列表解析中使用双 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]]]