gpt4 book ai didi

python - 在 Python 中的稀疏矩阵中按行查找非零条目的快速方法

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

我试图在稀疏矩阵中按行查找非零条目的索引:scipy.sparse.csc_matrix。到目前为止,我遍历矩阵中的每一行,并使用

numpy.nonzero()

到每一行以获取非零列索引。但是这种方法需要一个多小时才能找到每行的非零列条目。有快速的方法吗?谢谢!

最佳答案

使用 .nonzero() 方法。

indices = sp_matrix.nonzero()

如果您希望索引作为 (row, column) 元组,您可以使用 zip

indices = zip(*sp_matrix.nonzero())

关于python - 在 Python 中的稀疏矩阵中按行查找非零条目的快速方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/24790649/

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