From e6a00aabc86d57f5032714bf88dd1503d8d997bb Mon Sep 17 00:00:00 2001 From: Alex Lim Date: Tue, 4 Aug 2020 19:02:09 +0000 Subject: [PATCH] Using isnaobj instead of list comprehension (#35498) --- pandas/_libs/index.pyx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pandas/_libs/index.pyx b/pandas/_libs/index.pyx index ed98b52b461fa..b12c8830f15ab 100644 --- a/pandas/_libs/index.pyx +++ b/pandas/_libs/index.pyx @@ -29,7 +29,7 @@ from pandas._libs.tslibs.timedeltas cimport _Timedelta from pandas._libs.hashtable cimport HashTable from pandas._libs import algos, hashtable as _hash -from pandas._libs.missing import checknull +from pandas._libs.missing import checknull, isnaobj cdef inline bint is_definitely_invalid_key(object val): @@ -270,7 +270,7 @@ cdef class IndexEngine: Py_ssize_t i, j, n, n_t, n_alloc self._ensure_mapping_populated() - if any([checknull(t) for t in targets]): + if isnaobj(targets).any(): new_targets = [0 if checknull(t) else t for t in targets] new_values = [0 if checknull(v) else v for v in self._get_index_values()] targets = np.array(new_targets, dtype=object)