Python生成三对角矩阵

import numpy as np
def ThreediaMatrix(down,mid,up,n):         array_a=np.diag([down]*(n-1),-1)    # down是对角线下一行的元素，末尾是-1，n-1是元素的个数array_b = np.diag([mid] * n)        # mid是对角线上的元素，默认为0array_c=np.diag([up]*(n-1),1)       # up是对角线上面的元素，末尾是1matrix_A=array_a+array_b+array_creturn  matrix_A
A=ThreediaMatrix(-1,2,1,5)
print(A)[[ 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生成三对角矩阵相关推荐

python生成魔方矩阵
python生成魔方矩阵 import numpy as npdef magic(n):row, col = 0, n // 2magic = []for i in range(n):magic.ap ...
python 生成 Toeplitz 矩阵
python 生成 Toeplitz 矩阵首先导入必要的依赖文件 import numpy as np from scipy.linalg import toeplitz 明确要实现的功能 Toep ...
python 生成空白矩阵_3个用于数据科学的顶级Python库
用这些库把Python变成一个科学数据分析和建模工具. image by Opensource.com Python许多吸引人的特点如效率.代码可读性和速度使它成为数据科学爱好者的首选编程语言.对于希 ...
python生成正态分布矩阵_使用Numpy生成正态分布数据
如何使用Python生成正态分布的人员身高体重信息? 下面是生成的数据的分布情况,身高数据在各个区间大致呈正态分布. 对应的代码如下,先用pd.cut进行各个区间分桶,然后使用sort_index按照 ...
python生成任意n阶的三对角矩阵
数学作业要求实现共轭梯度法的算法. 题目中的矩阵A是n=400/500/600的三对角矩阵. 在网上查阅资料未果后,自己解决了. import numpy as npdef generate_matr ...
python meshgrid_numpy的生成网格矩阵 meshgrid()
numpy模块中的meshgrid函数用来生成网格矩阵,最简单的网格矩阵为二维矩阵 meshgrid函数可以接受 x1, x2,..., xn 等 n 个一维向量,生成 N-D 矩阵. 1 基本语法 ...
对称矩阵到三对角矩阵的Lanczos推导(python,数值积分)
第三十二篇 Lanczos转化到三对角形式在之前的篇章里,有许多求解线性方程的迭代方法,如最陡下降法,可以通过向量乘法和各种简单的向量运算,简化为一个单个矩阵的循环.将矩阵化为三对角形式的Lancz ...
python生成二维矩阵的方法
欢迎关注公众号--<数据三分钟> 一线大厂的师兄师姐结合自己的工作实践,将数据知识浅显道来,每天三分钟,助你成为数据达人.还有面试指导和内推机会. 1.直接用列表生成m行n列的矩阵 m,n ...
python生成零矩阵_python 实现矩阵填充0的例子
python 实现矩阵填充0的例子需求: 原矩阵 [[1 2 3] [4 5 6] [7 8 9]] 在原矩阵元素之间填充元素 0,得到 [[1. 0. 2. 0. 3.] [0. 0. 0. 0. ...

Python生成三对角矩阵

Python生成三对角矩阵相关推荐

最新文章

热门文章