python - 如何修剪 aa 列表并生成一个 numpy 二维数组？

Question

假设我有一个 numpy 数组：

[[1,2],
 [3,4],
 [5,6,7]]

是否有任何紧凑的方法来修剪阵列并使其沿第二维对齐，即

[[1,2],
 [3,4],
 [5,6]]

Answer 1

切片是否适用于 numpy 数组？如果是这样，下面的代码应该可以解决问题。

array = [[1,2],
[3,4],
[5,6,7]]

array = numpy.array([a[:2] for a in array])

我不确定切片是否适用于 numpy 数组，所以如果它是错误的，我将删除这个答案。