在每行的开头和结尾将 1 添加到 n 维 numpy 数组

Question

我有一个 NumPy 数组 data 的维度：

(100,2048)

我想在每行的开头添加 11 个，在末尾添加 11 个。

获取data.shape = (100, 11+2048+11)

我尝试了以下方法：

y = np.array([[1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0]])
z = np.append(y, data[0], axis=0)

它适用于第一个data[0]，方法是从左侧添加零。 但是，即使只有一行，我也无法在右侧添加。

我正在寻找类似的东西：

 z = np.append([y, data,y], axis=0)

例如：

z[0]= [1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,
.......................,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0]

..... 表示 2048 个值

Answer 1

你可以使用np.pad:

np.pad(z, ((0,0),(11,11)), 'constant', constant_values=1)

解释：((0,0), (11,11)) 表示填充到每个轴边缘的值的数量。在这里，我们在轴 0（行）之前和之后填充 0 个值，在每列之前和之后填充 11 个值。对于填充，我们使用常量值 1。

Answer 2

用零创建一个新数组并在需要的位置填充它：

offset = 11
z = np.ones((data.shape[0], data.shape[1] + offset * 2), dtype=data.dtype)
z[:, offset:z.shape[1] - offset] = data