1. shape
矩阵c ，c.shape[0]表示有几行，c.shape[1] 表示有几列
import numpy as np
c= np.array([[1,1],[1,2],[1,3],[1,4]])
print(c,"\n",c.shape[0]) #行数
print(c.shape[1]) #列数
输出：
[[1 1]
[1 2]
[1 3]
[1 4]]
4
2
2. numpy.eye() 生成对角矩阵
numpy.eye(N,M=None, k=0, dtype=<type 'float'>)
第一个参数：行数或列数
第二个参数: 列数，不写的话 默认生成方阵
第三个参数：默认情况下输出的是对角线全“1”，其余全“0”的方阵，如果k为正整数，则在右上方第k条对角线全“1”其余全“0”，k为负整数则在左下方第k条对角线全“1”其余全“0”。
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.]])
网友评论