用步长组成numpy数组

Question

我有一个数组a1_ijk，并希望将其广播到a2_ijmnk，但仅适用于m=n的情况。否则a2应该用零填充。当然，我可以创建新的空数组并用a1填充它，如下所示：

import numpy as np
A1 = np.random.rand(100, 5, 3)
A2 = np.zeros((100, 5, 2, 2, 3))
A2[..., 0, 0, :] = A1
A2[..., 1, 1, :] = A1

as_strided = np.lib.stride_tricks.as_strided
sz = A1.itemsize
A2 = as_strided(A1, shape=(100,5,2,2,3), strides=(5*3*sz, 3*sz, 0, 0, sz))

broadcast_to = np.lib.stride_tricks.broadcast_to
A2=broadcast_to(A1[...,None,None,:], (100,5,2,2,3))

不幸的是，这两个方法都用A1值填充所有m,n对。

我的问题是，如果这是可能的创建数组的形状我需要使用大步和没有实际复制数据？

Answer 1

我的问题是，这是否可能使用步幅来创建我需要的形状的数组，使用步幅而不实际复制数据？

才不是。作为一个简单的方法，考虑如果您的数组中没有零，那么该数组的任何视图都不会给出您所需的非对角线零。