gpt4 book ai didi

python - numpy合并上三角和下三角

转载 作者:行者123 更新时间:2023-12-01 19:33:19 25 4
gpt4 key购买 nike

我基本上想做与 this question 相反的事情.我有两个用 np.trilnp.triu 拆分的矩阵,我想将它们重新组合成一个矩阵。

A = array([[ 0. ,  0. ,  0. ],
[ 0.1, 0. , 0. ],
[ 0.6, 0.5, 0. ]])

B = array([[ 0. , 0.4, 0.8],
[ 0. , 0. , 0.3],
[ 0. , 0. , 0. ]])

我希望它看起来像

array([[ 0. ,  0.4,  0.8],
[ 0.1, 0. , 0.3],
[ 0.6, 0.5, 0. ]])

是否有内置的 numpy 函数来执行此操作?

最佳答案

你是说 A+B

import numpy
A = numpy.array([[ 0. , 0. , 0. ],
[ 0.1, 0. , 0. ],
[ 0.6, 0.5, 0. ]])

B = numpy.array([[ 0. , 0.4, 0.8],
[ 0. , 0. , 0.3],
[ 0. , 0. , 0. ]])

print A+B

返回

array([[ 0. ,  0.4,  0.8],
[ 0.1, 0. , 0.3],
[ 0.6, 0.5, 0. ]])

关于python - numpy合并上三角和下三角,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/22737757/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com