From e60d911511f14088eb068b303b2f720d3683295a Mon Sep 17 00:00:00 2001 From: Brock Date: Sun, 22 Nov 2020 10:46:17 -0800 Subject: [PATCH] REF: ensure_arraylike in algos.isin --- pandas/core/algorithms.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pandas/core/algorithms.py b/pandas/core/algorithms.py index be091314e6c25..a3abfaa48500c 100644 --- a/pandas/core/algorithms.py +++ b/pandas/core/algorithms.py @@ -218,7 +218,8 @@ def _ensure_arraylike(values): """ if not is_array_like(values): inferred = lib.infer_dtype(values, skipna=False) - if inferred in ["mixed", "string"]: + if inferred in ["mixed", "string", "mixed-integer"]: + # "mixed-integer" to ensure we do not cast ["ss", 42] to str GH#22160 if isinstance(values, tuple): values = list(values) values = construct_1d_object_array_from_listlike(values) @@ -424,6 +425,7 @@ def isin(comps: AnyArrayLike, values: AnyArrayLike) -> np.ndarray: values = construct_1d_object_array_from_listlike(list(values)) # TODO: could use ensure_arraylike here + comps = _ensure_arraylike(comps) comps = extract_array(comps, extract_numpy=True) if is_categorical_dtype(comps): # TODO(extension)