Python Pandas : Multi-Column Pivot and Level Swapping

Question

我有一个具有以下格式的初始数据框:

store_id,product,sale_ind,total_sold,percentage_sold
1,thing1,sale,30,46.2
1,thing2,no_sale,20,30.7
1,thing3,sale,15,23.1
2,thing4,sale,10,16.7
2,thing3,sale,20,33.3
2,thing2,sale,30,50.0
3,thing3,no_sale,20,50.0
3,thing2,sale,15,37.5
3,thing1,no_sale,5,12.5

我已经计算了我想要的一切，但现在我真的很难将这些数据 reshape 为以下格式:

                product
                sale_in
         total_sold percentage_sold
store_id
1,
2,
3,

当我尝试这个时:

df.pivot(index='store_id', columns='product')

我得到:ValueError:索引包含重复条目，无法 reshape 。

非常感谢任何提示!我担心我可能必须使用分层索引来解决问题。

Answer 1

您需要pivot_table来实现多列数据透视:

df.pivot_table(
    index=['store_id'], 
    columns=['product', 'sale_ind'], 
    values=['total_sold', 'percentage_sold']
)

或者在您的情况下，旋转时不涉及聚合，您可以使用 set_index 和 unstack:

df.set_index(['store_id', 'product', 'sale_ind']).unstack(['product', 'sale_ind'])