Extract numpy arrays from pandas dataframe as matrix

Question

I ended up with the following data structure:

import numpy as np
import pandas as pd

my_df = pd.DataFrame({'col_1': [1,2,3]})
my_df['col_1'] = my_df['col_1'].apply(lambda x: np.array([1,2,3]))
my_df.as_matrix()

it looks like this:

array([[array([1, 2, 3])],
       [array([1, 2, 3])],
       [array([1, 2, 3])]], dtype=object)

Whily I would like it to look like simply numpy matrix. Any thoughts?

piRSquared · Accepted Answer · 2017-10-16 16:22:57Z

2

If you just want an array, you can

np.array(my_df.col_1.tolist())

array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]])

answered Oct 16, 2017 at 16:22

piRSquared

296k68 gold badges509 silver badges654 bronze badges

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

1 Comment

user1700890 Over a year ago

Is it possible to apply it to multiple columns at once?

Collectives™ on Stack Overflow

Extract numpy arrays from pandas dataframe as matrix

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related