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.]])