如何从pandas框架中的特定列中提取numpy数组并将它们堆叠为单个numpy数组

时间:2017-05-10 00:33:56

标签: python pandas numpy

我有像这样的熊猫框架。

pd.DataFrame(data={'name':['name1','name2'],'vector':[np.array([1,2,3,4]),np.array([12,22,34,4])]})

我想从帧中提取矢量作为这样的矩阵。

np.array([[1,2,3,4],[12,22,34,4]])

2 个答案:

答案 0 :(得分:1)

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

将导致

array([[ 1,  2,  3,  4],
       [12, 22, 34,  4]])

df['vector'].as_matrix()

将导致

array([array([1, 2, 3, 4]), array([12, 22, 34,  4])], dtype=object)

答案 1 :(得分:0)

df.vector.values应该是最短的。