transform 1D array of list to 2D array

Question

For example

import pandas as pd
d1 = pd.Series(['a b', 'c d'])
t1 = d1.str.split()
a1 = t1.values

where a1 would be

array([list(['a', 'b']), list(['c', 'd'])], dtype=object)

how to transform it to

array([['a', 'b'],
       ['c', 'd']], dtype='<U1')

Andy L. · Accepted Answer · 2020-12-09 03:13:54Z

1

Use np.stack on t1:

In [186]: np.stack(t1)
Out[186]:
array([['a', 'b'],
       ['c', 'd']], dtype='<U1')

Or np.array on t1.tolist

In [187]: np.array(t1.tolist())
Out[187]:
array([['a', 'b'],
       ['c', 'd']], dtype='<U1')

answered Dec 9, 2020 at 3:13

Andy L.

25.3k4 gold badges20 silver badges30 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1