gpt4 book ai didi

Python:矩阵的非对角线元素为0

转载 作者:太空狗 更新时间:2023-10-30 02:54:00 25 4
gpt4 key购买 nike

将正方形对称 numpy ndarray 的非对角线元素转换为 0 的最快方法是什么?

最佳答案

我会检查保存对角线的速度,然后切换矩阵,然后恢复对角线:

n = len(mat)
d = mat.ravel()[::n+1]
values = d.copy()
mat[:,:] = 0
d[:] = values

如果矩阵不是很大,那么分配一个新矩阵可能会更快

mat = numpy.diag(numpy.diag(mat))

关于Python:矩阵的非对角线元素为0,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47648044/

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