2017-04-01 56 views
3

考虑:如何将字典转换为多级数据框?

A = pd.DataFrame([[1, 5, 2, 8, 2], [2, 4, 4, 20, 2], [3, 3, 1, 20, 2], [4, 2, 2, 1, 0], 
       [5, 1, 4, -5, -4], [1, 5, 2, 2, -20], [2, 4, 4, 3, 0], [3, 3, 1, -1, -1], 
       [4, 2, 2, 0, 0], [5, 1, 4, 20, -2]], 
      columns=['a', 'b', 'c', 'd', 'e'], 
      index=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) 

B = pd.DataFrame([[0, 0, 0, 8, 2], [1, 1, 1, 1, 1], [0, 0, 0, 8, 2], [0, 0, 2, 1, 0], 
       [5, 1, 4, -5, -4], [0, 0, 0, 8, 2], [2, 4, 4, 3, 0], [1, 3, 1, -1, -1], 
       [1, 1, 2, 0, 0], [2, 2, 2, 20, -2]], 
      columns=['a', 'b', 'c', 'd', 'e'], 
      index=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) 

test_list = [('test1', A), ('test2', B)] 
d_test = dict(test_list) 

然后,我遇到麻烦时,我尝试这本字典转换成一个多层次的数据帧。但我不知道如何指定多级别,我希望'key'是第二级数据框的名称。这是所需的输出

df 

    test_1    test_2 
    a b c d e a b c d e 
1 1 5 2 8 2 0 0 0 8 2 
2 2 4 4 20 2 1 1 1 1 1 
3 3 3 1 20 2 0 0 0 8 2 
4 4 2 2 1 0 0 0 2 1 0 
5 5 1 4 -5 -4 5 1 4 -5 -4 
6 1 5 2 2 -20 0 0 0 8 2 
7 2 4 4 3 0 2 4 4 3 0 
8 3 3 1 -1 -1 1 3 1 -1 -1 
9 4 2 2 0 0 1 1 2 0 0 
10 5 1 4 20 -2 2 2 2 20 -2 
+0

你需要一本字典吗? –

+0

此外,如果索引不是“B”或“A”会发生什么? –

+0

@WillemVanOnsem是的,除非从列表中更容易做到,否则我需要评估是否更改其余代码以适应此..... \\索引是相同的。 – hernanavella

回答

1

不仅仅是因为这里的答案 (其已经在评论中给出)的目的,这里有云(再次):

import pandas as pd 
A = pd.DataFrame([[1, 5, 2, 8, 2], [2, 4, 4, 20, 2], [3, 3, 1, 20, 2], [4, 2, 2, 1, 0], 
       [5, 1, 4, -5, -4], [1, 5, 2, 2, -20], [2, 4, 4, 3, 0], [3, 3, 1, -1, -1], 
       [4, 2, 2, 0, 0], [5, 1, 4, 20, -2]], 
      columns=['a', 'b', 'c', 'd', 'e'], 
      index=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) 
B = pd.DataFrame([[0, 0, 0, 8, 2], [1, 1, 1, 1, 1], [0, 0, 0, 8, 2], [0, 0, 2, 1, 0], 
       [5, 1, 4, -5, -4], [0, 0, 0, 8, 2], [2, 4, 4, 3, 0], [1, 3, 1, -1, -1], 
       [1, 1, 2, 0, 0], [2, 2, 2, 20, -2]], 
      columns=['a', 'b', 'c', 'd', 'e'], 
      index=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) 
AB=pd.concat([A,B], axis=1) 
header = ['test1','test1','test1','test1','test1','test2','test2','test2','test2','test2'] 
AB.columns = pd.MultiIndex.from_tuples(list(zip(header, AB.columns))) 
print(AB) # gives what was asked for 
print("test1: \n", AB.test1) # gives A 
print("test2: \n", AB.test2) # gives B 

另一个(下降到实生活状况)实现上述意见给予相同的(答案)的方式:

test_list = [('test1', A), ('test2', B)] 
d_test = dict(test_list) 
AB = pd.concat(d_test.values(), keys=d_test.keys(), axis=1) 
# what means: AB = pd.concat([A,B], keys=['test1', 'test2'], axis=1) 

上述输出码:

test1    test2    
     a b c d e  a b c d e 
1  1 5 2 8 2  0 0 0 8 2 
2  2 4 4 20 2  1 1 1 1 1 
3  3 3 1 20 2  0 0 0 8 2 
4  4 2 2 1 0  0 0 2 1 0 
5  5 1 4 -5 -4  5 1 4 -5 -4 
6  1 5 2 2 -20  0 0 0 8 2 
7  2 4 4 3 0  2 4 4 3 0 
8  3 3 1 -1 -1  1 3 1 -1 -1 
9  4 2 2 0 0  1 1 2 0 0 
10  5 1 4 20 -2  2 2 2 20 -2 
test1: 
    a b c d e 
1 1 5 2 8 2 
2 2 4 4 20 2 
3 3 3 1 20 2 
4 4 2 2 1 0 
5 5 1 4 -5 -4 
6 1 5 2 2 -20 
7 2 4 4 3 0 
8 3 3 1 -1 -1 
9 4 2 2 0 0 
10 5 1 4 20 -2 
test2: 
    a b c d e 
1 0 0 0 8 2 
2 1 1 1 1 1 
3 0 0 0 8 2 
4 0 0 2 1 0 
5 5 1 4 -5 -4 
6 0 0 0 8 2 
7 2 4 4 3 0 
8 1 3 1 -1 -1 
9 1 1 2 0 0 
10 2 2 2 20 -2