De uma matriz comodb (que será aproximadamente(1e6, 300)) e ummask = [1, 0, 1] vetor, eu defino o destino como 1 na primeira coluna.

Eu quero criar umout vetor que consiste naqueles em que a linha correspondente emdb corresponde aomask etarget==1e zeros em qualquer outro lugar.

db = np.array([       # out for mask = [1, 0, 1]
# target,  vector     #
  [1,      1, 0, 1],  # 1
  [0,      1, 1, 1],  # 0 (fit to mask but target == 0)
  [0,      0, 1, 0],  # 0
  [1,      1, 0, 1],  # 1
  [0,      1, 1, 0],  # 0
  [1,      0, 0, 0],  # 0
  ])

Eu defini umvline função que aplica ummask para cada linha da matriz usandonp.array_equal(mask, mask & vector) para verificar se os vetores 101 e 111 se encaixam na máscara, retém apenas os índices em quetarget == 1.

out é inicializado paraarray([0, 0, 0, 0, 0, 0])

out = [0, 0, 0, 0, 0, 0]

ovline A função é definida como:

def vline(idx, mask):
    line = db[idx]
    target, vector = line[0], line[1:]
    if np.array_equal(mask, mask & vector):
        if target == 1:
            out[idx] = 1

Eu obtenho o resultado correto aplicando esta função linha por linha em umfor ciclo:

def check_mask(db, out, mask=[1, 0, 1]):
    # idx_db to iterate over db lines without enumerate
    for idx in np.arange(db.shape[0]):
        vline(idx, mask=mask)
    return out

assert check_mask(db, out, [1, 0, 1]) == [1, 0, 0, 1, 0, 0] # it works !

Agora eu quero vetorizarvline criando umufunc:

ufunc_vline = np.frompyfunc(vline, 2, 1)
out = [0, 0, 0, 0, 0, 0]
ufunc_vline(db, [1, 0, 1])
print out

Mas oufunc reclama sobre a transmissão de entradas com essas formas:

In [217]:     ufunc_vline(db, [1, 0, 1])
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-217-9008ebeb6aa1> in <module>()
----> 1 ufunc_vline(db, [1, 0, 1])
ValueError: operands could not be broadcast together with shapes (6,4) (3,)
In [218]: