-
-
Notifications
You must be signed in to change notification settings - Fork 18.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
ENH(string dtype): Make str.decode return str dtype #60709
base: main
Are you sure you want to change the base?
Conversation
@@ -566,7 +566,7 @@ def test_string_slice_out_of_bounds(any_string_dtype): | |||
def test_encode_decode(any_string_dtype): | |||
ser = Series(["a", "b", "a\xe4"], dtype=any_string_dtype).str.encode("utf-8") | |||
result = ser.str.decode("utf-8") | |||
expected = ser.map(lambda x: x.decode("utf-8")).astype(object) | |||
expected = Series(["a", "b", "a\xe4"], dtype="str") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The change from ser.map
to using Series
is just to make this test a bit more explicit. Using ser.map(...).astype("str")
also passes.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good!
if get_option("future.infer_string"): | ||
data = ser.to_numpy() | ||
else: | ||
data = ser._values |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You can probably simplify this and always to .to_numpy()
? (or np.asarray(..)
)
In the case of object dtype in the else
branch, that will return the same (and be as cheap) as _values
I think
doc/source/whatsnew/vX.X.X.rst
file if fixing a bug or adding a new feature.