Scipy相当于numpy稀疏矩阵的位置

Question

我正在寻找numpy.where与scipy offers(scipy.sparse)的稀疏表示一起使用的等价物.是否有任何东西可以让你处理这些矩阵,就好像你在哪里使用if-then-else语句一样？

UPDATE 更具体:我需要where一个IF-THEN-ELSE vectorialized功能,即在这样的任务,因为这等于K的矩阵A的每个值,放在矩阵B对应的值,否则C.你可以使用类似find来检索满足逻辑条件的那些条目的索引,然后否定它们以找到所有剩余的条目,但对于稀疏矩阵,是不是有更紧凑的方法？

Answer 1

您可以使用scipy.sparse.find(http://docs.scipy.org/doc/scipy-0.9.0/reference/generated/scipy.sparse.find.html).此函数返回稀疏矩阵的非负值.对于某种情况,您可以使用ie:

 import scipy.sparse as sp
 A = sp.csr_matrix([[1, 2, 0], [0, 0, 3], [4, 0, 5]])
 B = A > 2 #the condition
 indexes = sp.find(B)