将向量作为Pandas数据帧元素

Question

有没有可能将一个向量作为Pandas数据帧的元素？

假设我有两个np向量：

a = np.array([[1,2,3],[4,5,6],[7,8,9]])
b = np.array([6, 15, 24])

那么，我想知道是否可以制作如下的一些2x3表格。

[1,2,3] | [4,5,6] | [7,8,9]
   6    |    15   |   24 

- noclew

Answer 1

你需要在 np.array 上使用 tolist 方法，以便让 pandas 根据你的需求来处理它。

pd.DataFrame(dict(a=a.tolist(), b=b)).T

- piRSquared

Answer 2

你可以将 a 和 b 作为列表的列表传递，无需转置。

In [2188]: pd.DataFrame([a.tolist(), b.tolist()])
Out[2188]:
           0          1          2
0  [1, 2, 3]  [4, 5, 6]  [7, 8, 9]
1          6         15         24

而且，更通用地说，

In [2202]: pd.DataFrame(x.tolist() for x in [a, b])
Out[2202]:
           0          1          2
0  [1, 2, 3]  [4, 5, 6]  [7, 8, 9]
1          6         15         24

- Zero