Propogating NaN values when using str.split (#18450) · pandas-dev/pandas@d64995a (original) (raw)

`@@ -2002,6 +2002,12 @@ def test_split_to_dataframe(self):

`

2002

2002

`5: [NA, 'not']})

`

2003

2003

`tm.assert_frame_equal(result, exp)

`

2004

2004

``

``

2005

`+

make sure we propogate NaN values across all columns

`

``

2006

`+

s = Series(["foo,bar,baz", NA])

`

``

2007

`+

result = s.str.split(",", expand=True)

`

``

2008

`+

exp = DataFrame([["foo", "bar", "baz"], [NA, NA, NA]])

`

``

2009

`+

tm.assert_frame_equal(result, exp)

`

``

2010

+

2005

2011

`s = Series(['some_splits', 'with_index'], index=['preserve', 'me'])

`

2006

2012

`result = s.str.split('_', expand=True)

`

2007

2013

`exp = DataFrame({0: ['some', 'with'], 1: ['splits', 'index']},

`