用Python创建一个Box矩阵

Question

我对python相对较新，我正在努力解决这个问题。它很简单，但我编码很糟糕。

写一个函数box_matrix（rows，cols），它创建一个具有给定行数和列数的新矩阵（2D NumPy数组），这样所有元素都是0，除了“edge”处的行和列矩阵的应该是1，（“空心”或“盒子”矩阵“）。

Answer 1

以0的矩阵开始，因为python中的索引从0开始，最后的行和列是行 - 1和cols - 1.索引中的：表示所有行或所有列。使用这两个索引事实用1填充“框”位置。

import numpy as np

def box_matrix(rows,cols):
    mat = np.zeros( (rows,cols ))
    mat[:,0] = 1
    mat[0,:] = 1
    mat[rows-1,:] = 1
    mat[:,cols-1] = 1
    return mat

bm = box_matrix(4,4)

print bm