Get DataFrame column value as array with target dtype

Question

I want to get the column values from DataFrame, which consists of arrays. By using DataFrame.values, the returned dtype is object, what I want is float64.

a=pd.DataFrame({'vector':[np.array([1.1,2,3]),np.array([2.1,3,4])]})
print(a)

b=a['vector'].values
print(b.dtype)
print(b.shape)

c=np.array([i for i in  a['vector']])
print(c.dtype)
print(c.shape)

>>>             vector
>>> 0  [1.1, 2.0, 3.0]
>>> 1  [2.1, 3.0, 4.0]
>>> object
>>> (2,)
>>> float64
>>> (2, 3)

why b and c has different dtype?

c is what I want to get, but is there any better way to get the same result?

Bharath M Shetty · Accepted Answer · 2017-12-27 07:05:49Z

1

Convert the Series to list and then pass it to np.array i.e

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

array([[ 1.1,  2. ,  3. ],
   [ 2.1,  3. ,  4. ]])

answered Dec 27, 2017 at 7:05

Bharath M Shetty

30.6k6 gold badges65 silver badges111 bronze badges

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

Comments

Hunger · Accepted Answer · 2017-12-29 09:22:59Z

0

According to https://stackoverflow.com/a/33718947/2251785,

numpy.concatenate should works too.

d=np.concatenate(a['vector'].values).reshape(len(a),-1)

Still confused about why .values treats array as object...

answered Dec 29, 2017 at 9:22

Hunger

5,4255 gold badges25 silver badges32 bronze badges

Collectives™ on Stack Overflow

Get DataFrame column value as array with target dtype

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related