python - 比较python中的两个scipy.sparse矩阵

Question

我在处理一个处理大型网络的程序，因此我必须使用稀疏矩阵(最好是 scipy.sparse.csr)。现在我想编写一个函数，它接受两个稀疏 bool 矩阵 A 和 B，并返回 B，但不包含 A 中设置的那些条目。这是一个伪代码示例。

def f(A, B):
    return B and (not A)

如何使用 scipy.sparse 矩阵完成此操作？

Answer 1

这是实现您的功能的一种方法:

def f(a, b):
    return b - b.multiply(a)

b.multiply(a) 是有效的元素级 and 操作。

这是一个例子。 a 和 b 是稀疏矩阵:

In [134]: b.A
Out[134]: array([[False, False,  True,  True]], dtype=bool)

In [135]: a.A
Out[135]: array([[False,  True, False,  True]], dtype=bool)

In [136]: f(a,b).A
Out[136]: array([[False, False,  True, False]], dtype=bool)