python - "append"多维 numpy 数组的正确方法？

Question

我有一个 4d 数组 x，我想循环遍历第一个轴，修改该 3d 数组，然后将此修改后的数组添加到新的 4d 数组 y >.

我目前正在做类似的事情:

xmod = modify(x[0, :, :, :])
y = xmod.reshape(1, x.shape[1], x.shape[2], x.shape[3])
for i in range(1, x.shape[0]):
    xmod = modify(x[i, :, :, :])
    y = np.vstack((y, xmod))

我猜有更多的清洁工可以做到这一点。怎么办？

Answer 1

如果您必须一次对 x 一个子矩阵执行操作，您可以这样做:

y = np.zeros_like(x)
for i in range(x.shape[0]):
    y[i,...] = modify(x[i,...])

例如

In [595]: x=np.arange(24).reshape(4,3,2)
In [596]: y=np.zeros_like(x)
In [597]: for i in range(x.shape[0]):
   .....:     y[i,...]=x[i,...]*2
   .....:     
In [598]: y
Out[598]: 
array([[[ 0,  2],
        [ 4,  6],
        ...
        [40, 42],
        [44, 46]]])

附加到列表通常比重复“附加”到数组更好:

In [599]: y=[]
In [600]: for row in x:
   .....:     y.append(row*2)
   .....: 
In [601]: y=np.array(y)

对于非常大的情况，您可以查看 vstack (或连接 axis=0)是否更快。但是您必须显式向数组添加起始维度。

In [615]: y=[]
In [616]: for row in x:
    y.append((row*2)[None,:])
   .....:     
In [617]: np.vstack(y)