Python 透视将字典作为列中的值的 DataFrame

发布于08月01日

我有一个数据帧，如下所示:

Code    Description Date    Quantity
dasd2132    dummy 1 2023-06 {'prev_value': 0.0, 'new_value': 90.0}
312dasda    dummy 2 2023-07 0
das432cs    dummy 3 2023-08 0
31231dsa    dummy 4 2023-09 0

它可以在其列中包含一个作为值的字典.

我希望透视此数据框，并将日期列值显示为列标题，将数量显示为列值:

Code    Description 2023-06                                 2023-07 2023-08 2023-09
dasd2132    dummy 1 {'prev_value': 0.0, 'new_value': 90.0}  0   0   0

有没有办法在Python语言中做到这一点？

请建议一个优化的解决方案. 谢谢.

我try 使用PIVOT_TABLE完成此操作:

pivoted_df = pd.pivot_table(df, index = ['Code', 'Description'], values = 'Quantity', columns = 'Date', sort = False)

但是，这会给出错误:TypeError: unhashable type: 'dict'

推荐答案

使用定制的pivot:

# identify non numeric values
m = pd.to_numeric(df['Quantity'], errors='coerce').isna()

out = (df  # keep only first Code/Description
   .assign(Code=lambda d: d['Code'].where(m).ffill(),
           Description=lambda d: d['Description'].where(m).ffill(),
           cols=lambda d: d.groupby(m.cumsum()).cumcount(),
          )
   .pivot(index=['Code', 'Description'], columns=['Date', 'cols'],
          values='Quantity')
   .droplevel(1, axis=1).reset_index().rename_axis(columns=None)
)

输出:

       Code Description                                 2023-06 2023-06 2023-06 2023-06
0  dasd2132     dummy 1  {'prev_value': 0.0, 'new_value': 90.0}       0       0       0