BUG: Series.str.split(expand=True) for ArrowDtype(pa.string())

[lukemanley]

• Tests added and passed if fixing a bug or adding a new feature
• All code checks passed.
• Added type annotations to new arguments/methods/functions.
• Added an entry in the latest doc/source/whatsnew/v2.0.3.rst file if fixing a bug or adding a new feature.

main:

In [1]: import pandas as pd

In [2]: import pyarrow as pa

In [3]: ser = pd.Series(["a", "a|b", "a|b|c"], dtype=pd.ArrowDtype(pa.string()))

In [4]: ser.str.split("|", expand=True)
Out[4]: 
   0
0  a
1  a
2  a

PR:

Out[4]: 
   0     1     2
0  a  <NA>  <NA>
1  a     b  <NA>
2  a     b     c

[mroeschke]

While we're here, could you change result._pa_array = (and the instanced above) to result = ArrowExtensionArray(...)? (so we can ensure the result._pa_array is always a pa.chunked_array?

[lukemanley]

yep, good catch. updated

[mroeschke]

Thanks @lukemanley

[lumberbot-app]

Owee, I'm MrMeeseeks, Look at me.

There seem to be a conflict, please backport manually. Here are approximate instructions:

1. Checkout backport branch and update it.

git checkout 2.0.x
git pull

2. Cherry pick the first parent branch of the this PR on top of the older branch:

git cherry-pick -x -m1 66468cecba6a1b17d2e12ff79da0b87f16118726

3. You will likely have some merge/cherry-pick conflict here, fix them and commit:

git commit -am 'Backport PR #53532: BUG: Series.str.split(expand=True) for ArrowDtype(pa.string())'

4. Push to a named branch:

git push YOURFORK 2.0.x:auto-backport-of-pr-53532-on-2.0.x

5. Create a PR against branch 2.0.x, I would have named this PR:

"Backport PR #53532 on branch 2.0.x (BUG: Series.str.split(expand=True) for ArrowDtype(pa.string()))"

And apply the correct labels and milestones.

Congratulations — you did some good work! Hopefully your backport PR will be tested by the continuous integration and merged soon!

Remember to remove the Still Needs Manual Backport label once the PR gets merged.

If these instructions are inaccurate, feel free to suggest an improvement.