有效地将大量SciPy稀疏矩阵项设置为z

Question

我需要从一个SciPy稀疏矩阵中删除大量的条目。 目前，我将矩阵转换为DOK格式并单独分配 每个条目都指向0。你知道吗

m = m.todok()
for i, j in pruneme:
  m[i,j] = 0

这是非常缓慢的。你知道吗

有没有更快的办法？你知道吗

Answer 1

只要不添加新的非零，就可以有效地设置CSR稀疏数组的元素，只需为数组订阅元组：

i, j = zip(*pruneme) # assuming that pruneme is a python list
m[i, j] = 0.
m.eliminate_zeros()

这应该比构造两个数组快得多。你知道吗