iis服务器助手广告广告
返回顶部
首页 > 资讯 > 后端开发 > Python >Python实现生成对角矩阵和对角块矩阵
  • 141
分享到

Python实现生成对角矩阵和对角块矩阵

Python 官方文档:入门教程 => 点击学习

摘要

目录对角矩阵scipy中的函数numpy.diagflatnumpy.tri对角块矩阵对角矩阵 scipy中的函数 在scipy.linalg中,通过tri(N, M=None, k

对角矩阵

scipy中的函数

在scipy.linalg中,通过tri(N, M=None, k=0, dtype=None)可生成N×M对角矩阵,若M=None,则M默认为N。k表示矩阵中用1填充的次对角线个数。

print(tri(3,5,2,dtype=int))
'''
[[1 1 1 0 0]
 [1 1 1 1 0]
 [1 1 1 1 1]]
'''

在numpy中也提供了多种对角矩阵生成函数,包括diag, diagflat, tri, tril, triu等,

numpy.diagflat

diagflat用于生成对角矩阵,diag在diagflat基础上,添加了提取对角元素的功能,例如

>>> np.diagflat([1,2,3])
array([[1, 0, 0],
       [0, 2, 0],
       [0, 0, 3]])
>>> np.diag([1,2,3])
array([[1, 0, 0],
       [0, 2, 0],
       [0, 0, 3]])
>>> np.diag(np.ones([3,3])) #提取对角元素
array([1., 1., 1.])

numpy.tri

tri(M,N,k)用于生成M行N列的三角阵,其元素为0或者1,k用于调节0和1的分界线相对于对角线的位置,例如

>>> np.tri(3,5,1)
array([[1., 1., 0., 0., 0.],
       [1., 1., 1., 0., 0.],
       [1., 1., 1., 1., 0.]])
>>> np.tri(3,5,2)
array([[1., 1., 1., 0., 0.],
       [1., 1., 1., 1., 0.],
       [1., 1., 1., 1., 1.]])
>>> np.tri(3,5,3)
array([[1., 1., 1., 1., 0.],
       [1., 1., 1., 1., 1.],
       [1., 1., 1., 1., 1.]])

tril, triu可用于提取出矩阵的左下和右上的三角阵,其输入参数除了待提取矩阵之外,另一个参数与tri中的k相同。

x = np.arange(12).reshape(4,3)
>>> np.tril(x,-1)
array([[ 0,  0,  0],
       [ 3,  0,  0],
       [ 6,  7,  0],
       [ 9, 10, 11]])
>>> np.triu(x,-1)
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 0,  7,  8],
       [ 0,  0, 11]])

对角块矩阵

对于scipy.linalg.block_diag(A,B,C)而言,会生成如下形式矩阵

from scipy.linalg import *
import numpy as np
A = np.ones([2,2])
B = np.round(np.random.rand(3,3),2)
C = np.diag([1,2,3])
bd = block_diag(A,B,C)
print(bd)
'''
[[1.   1.   0.   0.   0.   0.   0.   0.  ]
 [1.   1.   0.   0.   0.   0.   0.   0.  ]
 [0.   0.   0.8  0.38 0.41 0.   0.   0.  ]
 [0.   0.   0.84 0.45 0.24 0.   0.   0.  ]
 [0.   0.   0.32 0.22 0.25 0.   0.   0.  ]
 [0.   0.   0.   0.   0.   1.   0.   0.  ]
 [0.   0.   0.   0.   0.   0.   2.   0.  ]
 [0.   0.   0.   0.   0.   0.   0.   3.  ]]
'''

其中

到此这篇关于python实现生成对角矩阵和对角块矩阵的文章就介绍到这了,更多相关Python矩阵内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网!

--结束END--

本文标题: Python实现生成对角矩阵和对角块矩阵

本文链接: https://www.lsjlt.com/news/209117.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑,方便收藏和打印~

下载Word文档
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作