将numpy矩阵转换为单列pandas

时间:2018-05-22 20:45:35

标签: python pandas numpy

让我们考虑将形状为3x3的矩阵写成numpy数组:

np.array([[1,2,3],[4,5,6],[7,8,9]])

目标是将先前的向量存储在具有单个列的pandas数据帧中。在下图中是目标的表示。

enter image description here

2 个答案:

答案 0 :(得分:2)

pd.DataFrame([[i] for i in np.array([[1,2,3],[4,5,6],[7,8,9]])])

答案 1 :(得分:1)

在数组上使用tolist作为字典中的值。

pd.DataFrame({0: np.array([[1,2,3],[4,5,6],[7,8,9]]).tolist()})

           0
0  [1, 2, 3]
1  [4, 5, 6]
2  [7, 8, 9]

或没有字典

pd.DataFrame(np.array([[1,2,3],[4,5,6],[7,8,9]])[:, None].tolist())

           0
0  [1, 2, 3]
1  [4, 5, 6]
2  [7, 8, 9]