40python数据分析numpy基础之diag处理矩阵对角线元素

1 python数据分析numpy基础之diag处理矩阵对角线元素

python的numpy库的diag(v,k=0)函数，以一维数组的形式返回方阵的对角线元素，或将一维数组转换为方阵（非对角线元素为0）。

方阵：方形矩阵，行数和列数相等的矩阵。如果一个矩阵有n行和n列，则称为方阵。

矩阵：矩形阵列，由相同类型元素按矩形网格排列组成的二维结构。矩阵有2个维度，行和列，m×n的矩阵有m行n列。

主对角线：从矩阵的左上角到右下角的对角线，它的元素具有相同的行索引和列索引。

次对角线：从矩阵的右上角到左下角的对角线，它的元素的行索引+列索引=矩阵大小-1，a_ij，i+j=n-1，n为矩阵的行数或列数。

用法

numpy.diag(v, k=0)

描述

numpy.diag(v,k=0)，是一个用于创建和处理对角线数组的函数，提取矩阵对角线元素构造一维数组，或根据一维数组构造方阵。

入参

v：必选，列表、元组、数组，表示要处理的数组；

k：可选，默认为0，表示对角线位置；

1.1 入参v

numpy.diag(v,k=0)的入参v为必选入参；

如果v是二维数组，返回k位置的对角线；

如果v是一维数组，返回一个v作为k位置对角线的方阵，非对角线元素为0；

>>> import numpy as np
# diag(v)创建和处理对角线元素的函数
# v为列表
>>> np.diag([[1,2,3],[4,5,6],[7,8,9]])
array([1, 5, 9])
# v为元组
>>> np.diag(((1,2,3),(4,5,6),(7,8,9)))
array([1