Pandas: Getting back original numpy array stored in DataFrame

Question

Consider the following code:

import numpy as np
import pandas as pd

a = np.array([[1,2],[3,4],[5,6],[7,8]])
k = pd.DataFrame({"a": list(a)})

I'd like to retrieve the original numpy array. However, when I call .values I get something different.

How can I get the original numpy array, that looks like this:

array([[1, 2],
       [3, 4],
       [5, 6],
       [7, 8]])

?
Thank you.

jezrael · Accepted Answer · 2021-01-07 12:34:20Z

2

You can create nested lists and convert to 2d array:

np.array(k['a'].tolist())

answered Jan 7, 2021 at 12:29

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

1 Answer 1