如何将嵌套清单切成两次？

Question

有了一个嵌套列表，例如：

ex_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

我需要能够将此列表切成薄片：

[[1, 2], [4, 5]]

我一直在尝试：

list(ex_list[:2][:2])

但这不起作用。我显然做错了什么，但是由于某种原因使用逗号都无法使用解决方案，因此无法找到解决方案。

Answer 1

您需要将元素分别切成外部列表；最好先进行外部列表以避免不必要的内部切片。

[inner[:2] for inner in ex_list[:2]]

Answer 2

您应该尝试使用理解：
尝试：

[i[:2] for i in ex_list[:2]]

代码：

ex_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
print([i[:2] for i in ex_list[:2]])

输出：

[[1, 2], [4, 5]]

Answer 3

是否使用numpy一个选项？

import numpy as np

ex_list = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(ex_list[:2,:2].tolist()) # [[1, 2], [4, 5]]

第一个：2切片外列表，第二个切片内列表中的每个列表。

Answer 4

您可以尝试地图

l = list(map(lambda lst: lst[:2], ex_list))

print(l)

[[1, 2], [4, 5], [7, 8]]

或使用itemgetter

from operator import itemgetter

l = list(map(itemgetter(slice(0,2)), ex_list))

print(l)

[[1, 2], [4, 5], [7, 8]]