python - sqrt 用于元素稀疏矩阵

Question

我有一个稀疏矩阵:

from scipy import sparse
a = sparse.diags([1,4,9],[-1,0,1],shape =(10,10),format ="csr")

我想取稀疏矩阵中每个元素的平方根我在互联网上查找，它说我可以使用 numpy.sqrt() 来实现这一点。但出现错误:

  b = numpy.sqrt(a)
  AttributeError: sqrt

我该怎么做？

Answer 1

注意，这将创建一个结果 numpy ndarray 而不是稀疏 csr 数组。

from scipy import sparse
a = sparse.diags([1,4,9],[-1,0,1],shape =(10,10),format ="csr")

numpy.sqrt(a.data)

据我所知，除了 sqrt 之外，大多数其他 ufunc 运算(sin、cos、...)确实具有稀疏 ufunc，但不知道原因。请参阅此问题:https://github.com/scipy/scipy/pull/208