python - NumPy:将一维数组转换并 reshape 为带有零的二维数组？

Question

如何将以下列表转换并 reshape 为带零的二维数组？

# original list 
[1, 0.96, 0.92, 0.88]

# 2D Array
[[1     0     0     0   ]
 [0.96  1     0     0   ]
 [0.92  0.96  1     0   ]
 [0.88  0.92  0.96  1   ]]

Answer 1

这是一种时髦的矢量化方式。我们可以利用 np.lib.stride_tricks.as_strided 基于 scikit-image's view_as_windows 获得滑动窗口 View 并解决它。

from skimage.util.shape import view_as_windows

# a is input array (convert to array with np.array() is input is list)
p = np.r_[a[::-1], np.zeros(len(a)-1, dtype=a.dtype)]
out = view_as_windows(p,len(a))[::-1]

或者，保持它原生于 NumPy -

m = len(a)
n = p.strides[0]
out = np.lib.stride_tricks.as_strided(p[m-1:], shape=(m,m), strides=(-n,n))