2016-04-28 37 views
3

這是一個簡單的問題: 我有一個numpy矩陣A,我想轉換爲數據框(一個Series)中的一列,保留矩陣的多元索引。由此我的意思是,矩陣中的i,j位置將被轉換爲i,j在熊貓中進行索引。numpy matrix to 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 

有沒有辦法做到這一點?

回答

5

您可以使用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 
+0

這是真棒,謝謝! – tcapelle