在Python中创建上三角矩阵和下三角矩阵

Create upper and lower triangular matrix in python(在Python中创建上三角矩阵和下三角矩阵)
本文介绍了在Python中创建上三角矩阵和下三角矩阵的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

我想创建一个Python程序,该程序从一个具有一些系数的向量计算一个矩阵。这个矩阵是某种上下三角形。假设我们有以下系数向量a=[a0,a1,a2,a3,a4,a5],那么我想要计算矩阵:

我需要从向量a转到创建下/上三角矩阵A。我知道如何手动为其编制索引,但我需要一个可以执行此操作的程序。

我可能在考虑另一个循环中的循环,但我很难理解它在Python中是如何完成的,您认为这里应该做什么?

推荐答案

您可以尝试以下操作:

import numpy as np

c = [1, 2, 3, 4, 5, 6]
n = len(c)
m = 3
a = np.zeros((n,n))
for i in range(n):
    np.fill_diagonal(a[i:, :m], c[i])

np.fill_diagonal(a[:, m:], -1)    
print(a)

它提供:

[[ 1.  0.  0. -1.  0.  0.]
 [ 2.  1.  0.  0. -1.  0.]
 [ 3.  2.  1.  0.  0. -1.]
 [ 4.  3.  2.  0.  0.  0.]
 [ 5.  4.  3.  0.  0.  0.]
 [ 6.  5.  4.  0.  0.  0.]]

这篇关于在Python中创建上三角矩阵和下三角矩阵的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除!

相关文档推荐

Leetcode 234: Palindrome LinkedList(Leetcode 234:回文链接列表)
How do I read an Excel file directly from Dropbox#39;s API using pandas.read_excel()?(如何使用PANDAS.READ_EXCEL()直接从Dropbox的API读取Excel文件?)
subprocess.Popen tries to write to nonexistent pipe(子进程。打开尝试写入不存在的管道)
I want to realize Popen-code from Windows to Linux:(我想实现从Windows到Linux的POpen-code:)
Reading stdout from a subprocess in real time(实时读取子进程中的标准输出)
How to call type safely on a random file in Python?(如何在Python中安全地调用随机文件上的类型?)