Skip to content

Commit 1517f67

Browse files
committed
BUG: permit str dtype -> IntegerDtype conversions
Resolves pandas-dev#25472.
1 parent c021d33 commit 1517f67

File tree

2 files changed

+20
-15
lines changed

2 files changed

+20
-15
lines changed

pandas/core/arrays/integer.py

+13-8
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
is_integer_dtype,
3333
is_list_like,
3434
is_object_dtype,
35+
is_string_dtype,
3536
pandas_dtype,
3637
)
3738
from pandas.core.dtypes.missing import isna
@@ -119,17 +120,20 @@ def _get_common_dtype(self, dtypes: list[DtypeObj]) -> DtypeObj | None:
119120
return None
120121

121122

122-
def safe_cast(values, dtype, copy: bool):
123+
def safe_cast(values, dtype, inferred_type: str | None, copy: bool):
123124
"""
124125
Safely cast the values to the dtype if they
125126
are equivalent, meaning floats must be equivalent to the
126127
ints.
127-
128128
"""
129+
if inferred_type in ("string", "unicode"):
130+
# casts from str are always safe since they raise
131+
# a ValueError if the str cannot be parsed into an int
132+
return values.astype(dtype, copy=copy)
133+
129134
try:
130135
return values.astype(dtype, casting="safe", copy=copy)
131136
except TypeError as err:
132-
133137
casted = values.astype(dtype, copy=copy)
134138
if (casted == values).all():
135139
return casted
@@ -143,7 +147,7 @@ def coerce_to_array(
143147
values, dtype, mask=None, copy: bool = False
144148
) -> tuple[np.ndarray, np.ndarray]:
145149
"""
146-
Coerce the input values array to numpy arrays with a mask
150+
Coerce the input values array to numpy arrays with a mask.
147151
148152
Parameters
149153
----------
@@ -187,7 +191,8 @@ def coerce_to_array(
187191
return values, mask
188192

189193
values = np.array(values, copy=copy)
190-
if is_object_dtype(values):
194+
inferred_type = None
195+
if is_object_dtype(values) or is_string_dtype(values):
191196
inferred_type = lib.infer_dtype(values, skipna=True)
192197
if inferred_type == "empty":
193198
values = np.empty(len(values))
@@ -198,6 +203,8 @@ def coerce_to_array(
198203
"mixed-integer",
199204
"integer-na",
200205
"mixed-integer-float",
206+
"string",
207+
"unicode",
201208
]:
202209
raise TypeError(f"{values.dtype} cannot be converted to an IntegerDtype")
203210

@@ -230,9 +237,7 @@ def coerce_to_array(
230237
if mask.any():
231238
values = values.copy()
232239
values[mask] = 1
233-
values = safe_cast(values, dtype, copy=False)
234-
else:
235-
values = safe_cast(values, dtype, copy=False)
240+
values = safe_cast(values, dtype, inferred_type, copy=False)
236241

237242
return values, mask
238243

pandas/core/dtypes/common.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -143,14 +143,14 @@ def ensure_python_int(value: int | np.integer) -> int:
143143

144144

145145
def classes(*klasses) -> Callable:
146-
"""evaluate if the tipo is a subclass of the klasses"""
146+
"""Evaluate if the tipo is a subclass of the klasses."""
147147
return lambda tipo: issubclass(tipo, klasses)
148148

149149

150150
def classes_and_not_datetimelike(*klasses) -> Callable:
151151
"""
152-
evaluate if the tipo is a subclass of the klasses
153-
and not a datetimelike
152+
Evaluate if the tipo is a subclass of the klasses
153+
and not a datetimelike.
154154
"""
155155
return lambda tipo: (
156156
issubclass(tipo, klasses)
@@ -674,7 +674,7 @@ def is_integer_dtype(arr_or_dtype) -> bool:
674674
"""
675675
Check whether the provided array or dtype is of an integer dtype.
676676
677-
Unlike in `in_any_int_dtype`, timedelta64 instances will return False.
677+
Unlike in `is_any_int_dtype`, timedelta64 instances will return False.
678678
679679
The nullable Integer dtypes (e.g. pandas.Int64Dtype) are also considered
680680
as integer by this function.
@@ -726,7 +726,7 @@ def is_signed_integer_dtype(arr_or_dtype) -> bool:
726726
"""
727727
Check whether the provided array or dtype is of a signed integer dtype.
728728
729-
Unlike in `in_any_int_dtype`, timedelta64 instances will return False.
729+
Unlike in `is_any_int_dtype`, timedelta64 instances will return False.
730730
731731
The nullable Integer dtypes (e.g. pandas.Int64Dtype) are also considered
732732
as integer by this function.
@@ -1521,7 +1521,7 @@ def is_complex_dtype(arr_or_dtype) -> bool:
15211521

15221522
def _is_dtype(arr_or_dtype, condition) -> bool:
15231523
"""
1524-
Return a boolean if the condition is satisfied for the arr_or_dtype.
1524+
Return true if the condition is satisfied for the arr_or_dtype.
15251525
15261526
Parameters
15271527
----------
@@ -1580,7 +1580,7 @@ def get_dtype(arr_or_dtype) -> DtypeObj:
15801580

15811581
def _is_dtype_type(arr_or_dtype, condition) -> bool:
15821582
"""
1583-
Return a boolean if the condition is satisfied for the arr_or_dtype.
1583+
Return true if the condition is satisfied for the arr_or_dtype.
15841584
15851585
Parameters
15861586
----------

0 commit comments

Comments
 (0)