Question

我有两个数据框：

import pandas as pd
df1 = pd.DataFrame({'id': [1, 2,3],'param1': ['foo','bar','fu'],'param2': ['fo', 'ba','bar']})

   id param1 param2
0   1    foo     fo
1   2    bar     ba
2   3     fu    bar

df2 = pd.DataFrame({'id': [1, 1,1,2,2,3],'item': ['X','Y','Z','X','Z','Y'],})

   id item
0   1    X
1   1    Y
2   1    Z
3   2    X
4   2    Z
5   3    Y

我喜欢使用键id加入/合并两个帧。但是我希望框架能够扩展，因此每个独特的项目都会成为自己的列，并且每行都填充0或1.我猜它是一种带有枢轴的maping，也许？决赛桌看起来应该是这样的：

   id param1 param2  X  Y  Z
0   1    foo     fo  1  1  1
1   2    bar     ba  1  0  1
2   3     fu    bar  0  1  0

谢谢！

Answer 1

您可以将join与groupby一起使用并汇总size，然后将值转换为bool - 0至False和{{ 1}}到1,2...。最后的值转换为True - int和0：

crosstab的另一个解决方案：

print (df1.join(df2.groupby(['id', 'item'])
                   .size()
                   .unstack(fill_value=0)
                   .astype(bool)
                   .astype(int), on='id'))

   id param1 param2  X  Y  Z
0   1    foo     fo  1  1  1
1   2    bar     ba  1  0  1
2   3     fu    bar  0  1  0

Answer 2

你可以连接它们：

df_final = pd.concat([df1,df2],axis=1)
df_final = df_final.transpose().reset_index().drop_duplicates()
df_final = df_final.set_index('index').transpose()

输出：

    id  param1  param2  X   Y   Z
0   1   foo     fo      1   1   1
1   2   bar     ba      1   0   1
2   3   fu      bar     0   1   0

Python：使用False / True加入/合并带有pivot函数的两个panda数据帧

2 个答案: