How to extract numpy arrays from specific column in pandas frame and stack them as a single numpy array
本问题已经有最佳答案,请猛点这里访问。
我有一个像这样的熊猫架。
1 | pd.DataFrame(data={'name':['name1','name2'],'vector':[np.array([1,2,3,4]),np.array([12,22,34,4])]}) |
我想从框架中提取向量作为这样的矩阵。
1 | np.array([[1,2,3,4],[12,22,34,4]]) |
1 | np.array(df['vector'].tolist()) |
将导致
1 2 | array([[ 1, 2, 3, 4], [12, 22, 34, 4]]) |
或
1 | df['vector'].as_matrix() |
将导致
1 | array([array([1, 2, 3, 4]), array([12, 22, 34, 4])], dtype=object) |