python - 从嵌套列表中的子列表中提取第一个和最后一个元素

Question

如何从嵌套列表中的每个子列表中提取第一个和最后一个元素？
例如:

x=[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
print(x)
#[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

想要的结果——

[[1, 3], [4, 6], [7, 9]]

这是我见过的最接近的——

a = [x[i][i] for i in (0, -1)]
print(a)
#[1, 9]

Answer 1

你有正确的想法。如果您的列表中的列表只有一层深，您可以使用列表理解和 -1 索引访问第一个和最后一个元素，就像您尝试做的那样:

a = [[sublist[0],sublist[-1]] for sublist in x]

输出:

>>> a
[[1, 3], [4, 6], [7, 9]]