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

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]]

Answer 2

以下是如何使用列表切片：x = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] x = [l[::len(l)-1] for l in x] print(x)输出：[[1, 3], [4, 6], [7, 9]]

Answer 3

x = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]>>> [a[:1] + a[-1:] for a in x][[1, 3], [4, 6], [7, 9]]我提取了 2 个切片，一个包含第一个元素，一个包含最后一个元素，然后将它们连接起来。即使子列表的长度不同，它也会起作用。

Answer 4

您可以遍历列表并从那里访问它们：x=[[1, 2, 3], [4, 5, 6], [7, 8, 9]]out = []for arr in x:    out.append([arr[0], arr[-1]])print(out)输出：[[1, 3], [4, 6], [7, 9]]此外，您可以使用列表理解：out = [[arr[0], arr[-1]] for arr in x]

Answer 5

我建议这样做。该list[-1]语法旨在获取您想要的列表中的最后一个元素。x=[[1, 2, 3], [4, 5, 6], [7, 8, 9]]for elem in x:    print(elem[0], elem[-1])这只是以标准格式打印它们，但按此顺序将它们放回新列表中会很简单。

Answer 6

这有效：x = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]for item in x:    print(item[0], item[2])对于您想要的确切结果，请使用：x = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]new_list = []for item in x:    new_list.append([item[0], item[2]])print(new_list)