numpy matrix to pandas Series

时间:2016-04-28 13:59:19

标签: python numpy pandas

这是一个简单的问题: 我有一个numpy矩阵A,我想转换为数据框(系列)中的列,保留矩阵的多索引。我的意思是,矩阵中的i,j位置将被转换为i,j在pandas中的行索引。

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

有办法做到这一点吗?

1 个答案:

答案 0 :(得分:6)

您可以将stackDataFrame构造函数一起使用:

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