Numpy 2D阵列切片每行的不同元素

Question

假设我有一个2D numpy数组，

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

我可以将其切成薄片以选择每行2个元素：

a[:,0:2]

Output:
[[1,2]
 [4,5]]

但是我如何单独切成不同的行，例如第一行的3个元素，第二行元素是两个元素

Answer 1

您可以通过：创建两个单独的1D数组，

a[0][:3], a[1][:2]

但是您不能用两个不同尺寸的1D数组制作2D数组。请注意，Numpy中的数组是矩阵，而不是诸如Python列表的2D列表。

另一方面，您可以得到不同的切片，但每行的大小相同，并将它们作为2D数组。例如，假设您具有以下2D数组：

a = np.array([[1, 2, 3, 4, 5],[6, 7, 8, 9, 10]])

现在您可以从第一行中选择三个元素，第二行中的三个元素，但处于不同的位置：

a[0][:3], a[1][2:5]

然后您可以组合它们以制作新的2D数组：

np.vstack((a[0][:3], a[1][2:5]))

结果将是：

array([[ 1,  2,  3],
       [ 8,  9, 10]])

如果要替换列和行，则可以使用transpose或t：

np.vstack((a[0][:3], a[1][2:5])).T

那么您

array([[ 1,  8],
       [ 2,  9],
       [ 3, 10]])

• 也可以 使用：这两个阵列的组合：

np.hstack((a[0][:3], a[1][2:5]))

结果是：

array([ 1,  2,  3,  8,  9, 10])