在python中100 x 100的矩阵中,填充对角线元素

概述我正在尝试填充100×100矩阵上的对角线元素,如下面的matlab代码所示,所以如何在 python中复制它. T=(2*t0*diag(ones(1,100)))-(t0*diag(ones(1,99),1))-(t0*diag(ones(1,99),-1)) 所以我知道RHS的第一个项将填充矩阵的对角线,值为2 * t0, 我在python中做如下： x = np.zeros((100,10 我正在尝试填充100×100矩阵上的对角线元素,如下面的matlab代码所示,所以如何在 python中复制它.

T=(2*t0*diag(ones(1,100)))-(t0*diag(ones(1,99),1))-(t0*diag(ones(1,-1))

所以我知道RHS的第一个项将填充矩阵的对角线,值为2 * t0,

我在python中做如下：

x = np.zeros((100,100))np.fill_diagonal(x,2*t0)

但我不知道如何做第二和第三项,我知道它们会填充对角线元素上方和下方的值,值为-t0,不是所有的对角线值,而只是填充上下值对角线与-t0,休息都是零,但我不知道如何编写它的python代码.

我找到了这段代码：

# diagonal with offset from the main diagonaldiag([1,2,3],k=1)

将输出为：

array([[0,1,0],[0,0]])

但是如何在我的问题的情况下将它应用于大矩阵？
我在交互式python中工作,即Anaconda,所以我可以使用其他包来解决我的问题？

解决方法 从你在这里提供的np.diag,很容易做到：

a = np.ones((1,100))[0]b = np.ones((1,99))[0]m = np.diag(a,0) + np.diag(b,-1) + np.diag(b,1)

这里是100×100三对角矩阵

更新：

我在类似的问题上找到了链接here,看看也是如此.

总结

以上是内存溢出为你收集整理的在python中100 x 100的矩阵中,填充对角线元素全部内容，希望文章能够帮你解决在python中100 x 100的矩阵中,填充对角线元素所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。