python - 如何用不同长度的行对 numpy 数组进行切片

Question

我有一个 numpy 数组 arr，每个数组的长度各不相同。我可以得到arr的形状:

arr.shape

>>> (9,) 

我可以获得arr元素之一的形状:

arr[0].shape

>>> (6, 1, 2)

我知道所有这些元素的形状都是(n, 1, 2)。

我想对 arr 进行切片以获得一维结果，如下所示:

arr[:,:,:,0]

但我收到以下错误:

IndexError: too many indices for array

编辑:我原来的问题具有误导性。我想要执行此切片，以便可以为该切片分配值。因此，将切片放入新变量中对于我的情况没有用。本质上我想在一个简单的一行中做这样的事情:

arr[:,:,:,0] = arr[:,:,:,0] - np.min(arr[:,:,:,0])

Answer 1

您可以修复您的第一个(实际上是所有不同的)尺寸，并对 arr 的每个静态形状元素应用转换

import numpy as np
from random import randint

arr=np.array([np.random.randint(3,15, size=(randint(3,9),randint(3,7),randint(6,19))) for el in range(9)])

print(arr.shape)
print(arr[0].shape)
for i in range(arr.shape[0]):
    arr[i][:,:,0]-=arr[i][:,:,0].min()
    print(arr[i][:,:,0])