Python:将维度附加到二维数组

Question

假设你有一个数组 (m, m) 并且想把它变成 (n, n)。例如，将 2x2 矩阵转换为 6x6。所以:

[[ 1.  2.]
 [ 3.  4.]]

收件人:

[[ 1.  2.  0.  0.  0.  0.]
 [ 3.  4.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.]]

这就是我正在做的:

def array_append(old_array, new_shape):
    old_shape = old_array.shape
    dif = np.array(new_shape) - np.array(old_array.shape)
    rows = []
    for i in xrange(dif[0]):
        rows.append(np.zeros((old_array.shape[0])).tolist())
    new_array = np.append(old_array, rows, axis=0)
    columns = []
    for i in xrange(len(new_array)):
        columns.append(np.zeros(dif[1]).tolist())
    return np.append(new_array, columns, axis=1)

使用示例:

test1 = np.ones((2,2))
test2 = np.zeros((6,6))
print array_append(test1, test2.shape)

输出:

[[ 1.  1.  0.  0.  0.  0.]
 [ 1.  1.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.]]

基于 this回答。但这是一个(imho)简单操作的大量代码。有没有更简洁/pythonic 的方式来做到这一点？

Answer 1

为什么不使用array = numpy.zeros((6,6))，参见numpy docs ...

编辑，哎呀，问题已被编辑...我猜你是想把 ones 放在一个数组的一部分，里面全是 0？然后:

array = numpy.zeros((6,6))
array[0:2,0:2] = 1

如果小矩阵不都是1的值:

array[ystart:yend,xstart:xend] = smallermatrix