python如何生成任意n阶的三对角矩阵

 更新时间:2022年05月11日 09:30:43   作者:潮一  
这篇文章主要介绍了python如何生成任意n阶的三对角矩阵,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教

如何生成任意n阶的三对角矩阵

数学作业要求实现共轭梯度法的算法。

题目中的矩阵A是n=400/500/600的三对角矩阵。

在网上查阅资料未果后,自己解决了。

import numpy as np
def generate_matrix(n):
    # 使用对角矩阵相加得到三对角矩阵A
    array_a = np.diag([-2] * n)
    array = np.diag([1] * (n-1))
    a = np.zeros((n-1))
    b = np.zeros(n)
    array_b = np.insert(array, 0, values=a, axis=0)# 添加行
    array_b = np.insert(array_b, (n-1), values=b, axis=1)# 添加列
    array_c = np.insert(array, (n-1), values=a, axis=0)
    array_c = np.insert(array_c, 0, values=b, axis=1)
    matrix_A = array_a + array_b + array_c
    print(array_a)
    print(array_b)
    print(array_c)
    print(matrix_A)

# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    n = 5
    generate_matrix(n)

结果输出:

C:\Users\87167\Anaconda3\envs\tf\python.exe C:/Users/87167/PycharmProjects/GongETiDuFa/main.py
[[-2  0  0  0  0]
 [ 0 -2  0  0  0]
 [ 0  0 -2  0  0]
 [ 0  0  0 -2  0]
 [ 0  0  0  0 -2]]
[[0 0 0 0 0]
 [1 0 0 0 0]
 [0 1 0 0 0]
 [0 0 1 0 0]
 [0 0 0 1 0]]
[[0 1 0 0 0]
 [0 0 1 0 0]
 [0 0 0 1 0]
 [0 0 0 0 1]
 [0 0 0 0 0]]
[[-2  1  0  0  0]
 [ 1 -2  1  0  0]
 [ 0  1 -2  1  0]
 [ 0  0  1 -2  1]
 [ 0  0  0  1 -2]]

python矩阵分成上三角下三角和对角三个矩阵

diagonal

Return specified diagonals.

diagflat

Create a 2-D array with the flattened input as a diagonal.

trace

Sum along diagonals.

triu

Upper triangle of an array.

tril

Lower triangle of an array.

先讲一个方阵的对角线下的下三角阵和对角线上的上三角阵提取出来(如果只需要上下三角阵,则去掉tril/triu中的第二个参数)

上代码(这里使用tril和triu都是返回array形式,还需使用mat转换回矩阵):

>>> m = np.mat("1,2,3;4,5,6;7,8,9")
>>> m
matrix([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])
>>> L = np.tril(m,-1)
>>> L
array([[0, 0, 0],
       [4, 0, 0],
       [7, 8, 0]])
>>> U = np.triu(m,1)
>>> U
array([[0, 2, 3],
       [0, 0, 6],
       [0, 0, 0]])

而单独要提取对角线上的元素作为一个矩阵有如下两种方法:

1、运用np.diag两次,再使用mat转换回矩阵:

>>> D = np.diag(np.diag(m))
>>> D
array([[1, 0, 0],
       [0, 5, 0],
       [0, 0, 9]])
>>> D = np.mat(D)
>>> D
matrix([[1, 0, 0],
        [0, 5, 0],
        [0, 0, 9]])

2、运用下三角矩阵减去次下三角矩阵(即对角线下的下三角阵):

>>> D = np.tril(m) - L
>>> D
array([[1, 0, 0],
       [0, 5, 0],
       [0, 0, 9]])
>>> D = np.mat(D)
>>> D
matrix([[1, 0, 0],
        [0, 5, 0],
        [0, 0, 9]])

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • Python闭包与装饰器原理及实例解析

    Python闭包与装饰器原理及实例解析

    这篇文章主要介绍了Python闭包与装饰器原理及实例解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-04-04
  • 简单了解python中的f.b.u.r函数

    简单了解python中的f.b.u.r函数

    这篇文章主要介绍了简单了解python中的f.b.u.r函数,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-11-11
  • Python中规范定义命名空间的一些建议

    Python中规范定义命名空间的一些建议

    命名空间是Python程序的一大根本,编程时持命名空间的整洁还是十分必要的,这里就来为大家总结Python中规范定义命名空间的一些建议,需要的朋友可以参考下
    2016-06-06
  • python使用xlrd与xlwt对excel的读写和格式设定

    python使用xlrd与xlwt对excel的读写和格式设定

    最近在用python处理excel表的时候出现了一些问题,所以想着记录下最后的实现方式和问题解决方法。方便自己或者大家在有需要的时候参考借鉴,下面这篇文章主要就介绍了python使用xlrd与xlwt对excel的读写和格式设定的相关资料,一起来学习学习吧。
    2017-01-01
  • pytest自动化测试数据驱动yaml/excel/csv/json

    pytest自动化测试数据驱动yaml/excel/csv/json

    这篇文章主要为大家介绍了pytest自动化测试数据驱动yaml/excel/csv/json的示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-06-06
  • 在Python 2.7即将停止支持时,我们为你带来了一份python 3.x迁移指南

    在Python 2.7即将停止支持时,我们为你带来了一份python 3.x迁移指南

    这篇文章主要介绍了在Python 2.7即将停止支持时我们为你准备了一份python 3.x迁移指南的相关资料,需要的朋友可以参考下
    2018-01-01
  • Python连接Redis的基本配置方法

    Python连接Redis的基本配置方法

    本文分享了Linux系统下Python连接Redis的基本配置方法,Python连接Redis需要Redis的Python包支持
    2018-09-09
  • 解决keras使用cov1D函数的输入问题

    解决keras使用cov1D函数的输入问题

    这篇文章主要介绍了解决keras使用cov1D函数的输入问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • python实现飞行棋游戏

    python实现飞行棋游戏

    这篇文章主要为大家详细介绍了python实现飞行棋游戏,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2020-02-02
  • python求两个时间的时间差(实例代码)

    python求两个时间的时间差(实例代码)

    我们在用python进行分析的时候,可能会碰到计算两个日期的时间差。下面为大家介绍一下如何计算两个时间的时间差,需要的朋友可以参考下
    2022-11-11

最新评论