File tree Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -272,9 +272,10 @@ cdef class IndexEngine:
272
272
self ._ensure_mapping_populated()
273
273
values = np.array(self ._get_index_values(), copy = False )
274
274
targets_mask = isnaobj(targets)
275
+ values_mask = isnaobj(values)
275
276
if targets_mask.any():
276
277
targets[targets_mask] = 0
277
- values[isnaobj(values) ] = 0
278
+ values[values_mask ] = 0
278
279
279
280
stargets = set (targets)
280
281
n = len (values)
@@ -325,8 +326,13 @@ cdef class IndexEngine:
325
326
n_alloc += 10 _000
326
327
result = np.resize(result, n_alloc)
327
328
328
- result[count] = j
329
- count += 1
329
+ if val == 0 :
330
+ if targets_mask[i] == values_mask[j]:
331
+ result[count] = j
332
+ count += 1
333
+ else :
334
+ result[count] = j
335
+ count += 1
330
336
331
337
# value not found
332
338
else :
You can’t perform that action at this time.
0 commit comments