pandas dataframe to numpy array without the annoying dtype="blah blah"

Question

I selected several rows of the first column and used both .values and as_matrix() to convert it to numpy array.

x1 = df.iloc[[1,3,4,64,34], 0]

The dimension of the each element in the dataframe is a (3, 3, 5) numpy array. I would like to output a (5, 3, 3, 5) numpy array from the dataframe. However when I use .values, i get a (5, ) shape array with a dtype =unit8 at the end of each element. How can I go about fixing this?

must have been by accident

user252046
– user252046

2017-06-01 21:08:17 +00:00
Commented Jun 1, 2017 at 21:08 — user252046
– user252046, Commented Jun 1, 2017 at 21:08
Thanks, appreciate it

piRSquared
– piRSquared

2017-06-01 21:08:53 +00:00
Commented Jun 1, 2017 at 21:08 — piRSquared
– piRSquared, Commented Jun 1, 2017 at 21:08

piRSquared · Accepted Answer · 2017-03-28 05:59:27Z

1

df = pd.DataFrame(dict(A=[np.ones((3, 3, 5))] * 5))

A = np.array([a.tolist() for a in df.iloc[[0, 1, 2, 3, 4], 0]])

print(A.shape)

(5, 3, 3, 5)

answered Mar 28, 2017 at 5:59

piRSquared

296k68 gold badges509 silver badges654 bronze badges

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

Collectives™ on Stack Overflow

pandas dataframe to numpy array without the annoying dtype="blah blah"

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related