Skip to content

Instantly share code, notes, and snippets.

import numpy as np
from scipy.sparse import dia_matrix
import timeit
def matches(a, b):
mask = dia_matrix((np.ones((1, a.size)).repeat(4, axis=0), np.arange(4)),
shape=(a.size, b.size), dtype=np.int)
matches = a[mask.T.dot(mask.dot(a == b) == 4).astype(np.bool)]
if __name__ == '__main__':