如何从熊猫框架中的特定列中提取numpy数组并将它们堆叠为单个numpy数组[重复]

Posted 2023-03-11

【中文标题】如何从熊猫框架中的特定列中提取numpy数组并将它们堆叠为单个numpy数组[重复]

【英文标题】：How to extract numpy arrays from specific column in pandas frame and stack them as a single numpy array [duplicate]

【发布时间】：2017-10-08 11:21:55

【问题描述】：

我有一个这样的熊猫框架。

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]])

【参考方案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)

【参考方案2】：

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