我有一个列表的列表,我想从每个子列表中提取第 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]]]
