重组python pandas中的表

时间:2017-06-30 00:40:22

标签: python-3.x pandas dataframe

我有两个表,它们是数据帧对象(pandas),如下所示:

thread.get_ident()

我想创建这样的东西:

import thread

1 个答案:

答案 0 :(得分:2)

我认为pandas.concat需要DataFrame.unstack

df1 = pd.DataFrame({'col3': [1, 3, 2, 1, 5, 0, 3, 4], 
                    'col2': ['a1', 'a2', 'a3', 'a1', 'a2', 'a1', 'a2', 'a3'], 
                    'col1': ['A', 'A', 'A', 'B', 'B', 'C', 'C', 'C']})
df1 = df1.set_index(['col1','col2'])
print (df1)
           col3
col1 col2      
A    a1       1
     a2       3
     a3       2
B    a1       1
     a2       5
C    a1       0
     a2       3
     a3       4

df2 = pd.DataFrame({'col3': [2, 1, 1, 2, 2, 0], 
                    'col2': ['b1', 'b2', 'b1', 'b2', 'b1', 'b2'], 
                    'col1': ['A', 'A', 'B', 'B', 'C', 'C']})
df2 = df2.set_index(['col1','col2'])
print (df2)
           col3
col1 col2      
A    b1       2
     b2       1
B    b1       1
     b2       2
C    b1       2
     b2       0
df = pd.concat([df1, df2])['col3'].unstack(fill_value=0)
print (df)
col2   a1 a2 a3 b1 b2
col1                 
A       1  3  2  2  1
B       1  5  0  1  2
C       0  3  4  2  0

最后如果需要删除列,索引名称会添加rename_axis

df = pd.concat([df1, df2])['col3'] \
       .unstack(fill_value=0) \
       .rename_axis(None) \
       .rename_axis(None, axis=1)
print (df)
   a1  a2  a3  b1  b2
A   1   3   2   2   1
B   1   5   0   1   2
C   0   3   4   2   0