Python：如何将独热编码后的DataFrame按brokerid聚合展平？

如果你已经完成了独热编码，直接用这行代码合并：

result_df = df.groupby('brokerid').max().reset_index()

解释 #

独热编码后，同一brokerid的多行数据里，只要该broker关联过某个insurer，对应列就会有至少一行值为1。groupby('brokerid').max()会把同一brokerid的所有行合并，对每列取最大值——这样关联过的insurer列会保留1，未关联的保持0。reset_index()是把brokerid从索引变回普通列，不需要的话可以去掉。

如果还没做独热编码，把两步合并成一行：

import pandas as pd
result_df = pd.get_dummies(df, columns=['insurer']).groupby('brokerid').max().reset_index()

内容的提问来源于stack exchange，提问作者Will_I_Amnt