我有两个表,它们是数据帧对象(pandas),如下所示:
thread.get_ident()
我想创建这样的东西:
import thread
答案 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