合并具有分层列的两个DataFrame

时间:2013-09-23 17:10:45

标签: python pandas

我想合并两个DataFrame,同时创建一个多级列命名方案,表示行来自哪个数据帧。例如:

In [98]: A=pd.DataFrame(np.arange(9.).reshape(3,3),columns=list('abc'))
In [99]: A
Out[99]: 
   a  b  c
0  0  1  2
1  3  4  5
2  6  7  8

In [100]: B=A.copy()

如果我使用pd.merge(),那么我会

In [104]: pd.merge(A,B,left_index=True,right_index=True)
Out[104]: 
   a_x  b_x  c_x  a_y  b_y  c_y
0    0    1    2    0    1    2
1    3    4    5    3    4    5
2    6    7    8    6    7    8

我对这个陈述的期望是什么,我想要的是什么(但我不知道怎么做!)是:

In [104]: <<one or more statements>>
Out[104]: 
     A              B
     a    b    c    a    b    c
0    0    1    2    0    1    2
1    3    4    5    3    4    5
2    6    7    8    6    7    8

是否可以在不更改原始pd.DataFrame来电的情况下完成此操作?我正在从.csv文件中读取数据框中的数据,这可能是我的问题。

2 个答案:

答案 0 :(得分:7)

第一种情况可以在A,B之间任意排序(不是列,只是A或B的顺序) 第二应保留订购

恕我直言,这是一个pandonic!

In [5]: concat(dict(A = A, B = B),axis=1)
Out[5]: 
   A        B      
   a  b  c  a  b  c
0  0  1  2  0  1  2
1  3  4  5  3  4  5
2  6  7  8  6  7  8

In [6]: concat([ A, B ], keys=['A','B'],axis=1)
Out[6]: 
   A        B      
   a  b  c  a  b  c
0  0  1  2  0  1  2
1  3  4  5  3  4  5
2  6  7  8  6  7  8

答案 1 :(得分:6)

这是一种改变A和B的方法:

In [10]: from itertools import cycle

In [11]: A.columns = pd.MultiIndex.from_tuples(zip(cycle('A'), A.columns))

In [12]: A
Out[12]:
   A
   a  b  c
0  0  1  2
1  3  4  5
2  6  7  8

In [13]: B.columns = pd.MultiIndex.from_tuples(zip(cycle('B'), B.columns))

In [14]: A.join(B)
Out[14]:
   A        B
   a  b  c  a  b  c
0  0  1  2  0  1  2
1  3  4  5  3  4  5
2  6  7  8  6  7  8

我实际上认为这是一个很好的替代行为,而不是后缀......