numpy 矩阵到熊猫系列

Question

这是一个简单的问题： 我有一个 numpy 矩阵 A ，我想将它转换为数据框（系列）中的一列，保留矩阵的多索引。我的意思是，矩阵中的 i,j 位置要转换为 pandas 中的 i,j 行索引。

In [68]: A = np.array([[1,2],[3,4]])

利用这个矩阵，一些巫术魔法，得到这个：

In [69]: Series([1,2,3,4],index = [[0,0,1,1],[0,1,0,1]])
Out[69]: 
0  0    1
   1    2
1  0    3
   1    4
dtype: int64

有没有办法做到这一点？

Answer 1

您可以将stack 与DataFrame 构造函数一起使用：

import pandas as pd
import numpy as np

print pd.DataFrame(np.array([[1,2],[3,4]])).stack()

0  0    1
   1    2
1  0    3
   1    4
dtype: int32