特牛网址导航

python生成任意n阶的三对角矩阵_打印n阶对角矩阵python-CSDN博客

网友收藏
数学作业要求实现共轭梯度法的算法。题目中的矩阵A是n=400/500/600的三对角矩阵。在网上查阅资料未果后,自己解决了。import numpy as npdef 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 = n_打印n阶对角矩阵python