如何用List comprehension提取嵌套列表的第2和第4个元素

原嵌套列表 #

mylist = [
    [2.1, 5.2, 6.1, -0.1],
    [2.2, 4.8, 5.6, -1.0],
    [3.3, 6.5, 5.9, 1.2],
    [2.9, 5.6, 6.0, 0.0],
]

期望生成的结果 #

result = [
    [5.2, -0.1],
    [4.8, -1.0],
    [6.5, 1.2],
    [5.6, 0.0]
]

解决方案 #

直接用列表推导式就能实现需求：

result = [[sublist[1], sublist[3]] for sublist in mylist]

逻辑说明 #

遍历mylist里的每一个子列表sublist，对每个子列表提取**索引1（第二个元素）和索引3（第四个元素）**的值，把这两个值组成新的子列表，所有新子列表会自动汇总成最终的result列表，完全匹配目标输出。

内容的提问来源于stack exchange，提问作者missingfours