Question

我正在尝试使用带有csr_matrix的numpy.where，这不起作用。我问的是有一些内置函数等效于稀疏矩阵的numpy.where。以下是我不想使用 Forloop 或 .todense（）

的示例

 import scipy.sparse as spa
 import numpy as np
 N = 100
 A = np.zeros((N,N))

 di = np.diag_indices((len(A[:,0])))
 A[di] = 2.3
 '''
 adding some values to non-diagonal terms 
 for sake of example 
 '''
 for k in range(0,len(A)-1):
     for j in range(-1,3,2):
         A[k,k+j] = 4.0
 A[2,3] =0.1
 A[3,3] = 0.1
 A[0,4] = 0.2
 A[0,2] = 3

 '''
 creating sparse matrix
 '''
 A = spa.csc_matrix((N,N))
 B = spa.csc_matrix((N,N))

 '''
 Here I get
 TypeError: unsupported operand type(s) for &: 'csc_matrix' and 'csc_matrix'
 '''

 ind1 = np.where((A>0.0) & (A<=1.0)) 
 B[ind1] = (3.0-B[ind1])**5-6.0*(2.0-B[ind1])**5

Answer 1

如何使用A和B的基础数组，data数组

In [36]: ind2=np.where((A.data>0.0)&(A.data<=1.0))

In [37]: A.indices[ind2]
Out[37]: array([2, 3, 0])

In [38]: A.indptr[ind2]
Out[38]: array([28, 31, 37])

In [39]: A.data[ind2]
Out[39]: array([ 0.1,  0.1,  0.2])

In [41]: B.data[ind2]=(3.0-B.data[ind2])**5-6.0*(2.0-B.data[ind2])**5

In [42]: B.data[ind2]
Out[42]: array([ 56.54555,  56.54555,  58.7296 ])

要查看密集版本中ind2对应的内容，请将数组转换为coo

In [53]: Ac=A.tocoo()

In [54]: (Ac.row[ind2], Ac.col[ind2])
Out[54]: (array([2, 3, 0]), array([3, 3, 4]))

其中，作为参考，密集阵列上的where为：

In [57]: np.where((A.A>0.0) & (A.A<=1.0))
Out[57]: (array([0, 2, 3]), array([4, 3, 3]))

一个重要的注意事项 - 使用A.data意味着排除密集阵列的所有零条目。

numpy.where相当于csr_matrix python

1 个答案: