通过一次删除一项从向量创建一个 numpy 数组

【问题标题】：Creating a numpy array from a vector by removing one item at a time通过一次删除一项从向量创建一个 numpy 数组
【发布时间】：2018-07-11 13:33:18
【问题描述】：

我有一个大小为 n 的元素列表 b。我想从此列表中创建一个大小为 (n-1,n) 的 2D numpy 数组，这样第 i 行是 b 的元素，没有第 i 个项。

例如，如果

b = [1,2,3,4]

numpy 数组将是，

A = np.array( [[2,3,4],
               [1,3,4],
               [1,2,4],
               [1,2,3]])

【问题讨论】：

标签： arrays python-2.7 numpy

【解决方案1】：

方法 #1： 一种使用 masking 的方法 -

n = len(b)
b2D = np.broadcast_to(b, (n,n)) # or np.repeat(b[None],n,axis=0)
out = b2D[~np.eye(n, dtype=bool)].reshape(n,-1)

方法#2：关注性能和内存效率，另一个关注NumPy strides -

strided = np.lib.stride_tricks.as_strided
n = len(b)
b_ext = np.r_[b[1:],b[:-1]]
s = b_ext.strides[0]
out = strided(b_ext, shape=(n-1,n), strides=(s,s)).reshape(n,-1)

【讨论】：