将矢量作为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 

Answer 1

您需要使用tolist上的np.array方法让熊猫按照您的喜好进行操作。

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

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