用python 替换2D数组的对角线

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了用python 替换2D数组的对角线，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1350字，纯文字阅读大概需要2分钟。

内容图文

参见英文答案 > changing the values of the diagonal of a matrix in numpy                                    7个
我有以下2D数组

A=([[1, 2, 3, 4],
    [5, 6, 7, 8],
    [9, 10, 11, 12],
    [13, 14, 15, 16])

我想用数组替换主对角线

a = ([0,2,15,20])

因此,结果必须如此

A=([[0, 2, 3, 4],
    [5, 2, 7, 8],
    [9, 10, 15, 12],
    [13, 14, 15, 20])

我尝试使用np.diag(a,k = 0),但它不起作用,因为np.diag()创建了一个带有数组“a”的对角线2D数组.

有没有办法用numpy做到这一点？
上面的例子是最简单的例子.我希望能够不仅改变邮件对角线而且改变所有对角线.

解决方法:

您可以使用np.fill_diagonal(..).就像文档说的那样：

numpy.fill_diagonal(a, val, wrap=False)

Fill the main diagonal of the given array of any dimensionality.

例如：

np.fill_diagonal(A, 20)

因此,我们在整个对角线上播放了20个.

您还可以使用不同的值填充对角线,例如：

np.fill_diagonal(A, [0,2,15,20])

例如：

>>> a = np.zeros((4,4))
>>> np.fill_diagonal(a, [0,2,15,20])
>>> a
array([[ 0.,  0.,  0.,  0.],
       [ 0.,  2.,  0.,  0.],
       [ 0.,  0., 15.,  0.],
       [ 0.,  0.,  0., 20.]])

如果您想要更改其他对角线,则需要镜像阵列.例如对于反对角线,我们可以使用：

np.fill_diagonal(A[::-1], -20)

然后我们得到：

>>> A = np.zeros((4,4))
>>> np.fill_diagonal(A[::-1], -20)
>>> A
array([[  0.,   0.,   0., -20.],
       [  0.,   0., -20.,   0.],
       [  0., -20.,   0.,   0.],
       [-20.,   0.,   0.,   0.]])

如果我们不考虑超对角线和子对角线,则n维矩阵具有n×(n-1)个对角线.我们可以通过镜像一个或多个维度来分配它.

内容总结

以上是互联网集市为您收集整理的用python 替换2D数组的对角线全部内容，希望文章能够帮你解决用python 替换2D数组的对角线所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端