如何将子列表切片到给定的长度？

Question

我的问题是列表的子列表，例如l = [[1,2,3,4], [5,6,7,8],[9,10,11,12]] 将被切片 到给定的长度，比如 3。结果应该是这样的：l = [[1,2,3], [5,6,7],[9,10,11]]。 当然你可以这样做：

l = [i[:3] for i in l]

但我想知道：是否有更短的方法，即仅使用切片运算符并避免 for 循环？

Answer 1

如果您使用 numpy 并使 l 成为一个 ndarray (np.array(l))，您可以这样做：

l = l[:,:3]

这意味着：所有行和每行从 0 开始取 3 列。