np.identity和np.eye
numpy.identity(n, dtype=None) >>> np.identity( 3 ) array( [[ 1., 0., 0.], [ 0., 1., 0.], [ 0., 0., 1.]] ) numpy.eye(N, M=None, k=0, dtype=< class ‘float’>, order=’C’) Parameters: N : int 行数 M : int, 可选,列数,默认等于N k : int, 可选,对角线索引:0(默认值)是指主对角线,正值是指上对角线,而负值是指向下对角线。 order : {‘C’, ‘F’}, optional 可选,内存中以行存储还是以列存储。 >>> np.eye( 2 , dtype=int) array( [[1, 0], [0, 1]] ) >>> np.eye( 3 , k= 1 ) array( [[ 0., 1., 0.], [ 0., 0., 1.], [ 0., 0., 0.]] ) 相当于k等于1,对角线就整体往右边移动一位。 文章来源: np.identity和np.eye