根据 pivot_table 的输出动态重命名数据框

Question

我试图以一种格式压缩我的数据，这样它会更有用，数据格式：

Table1

Key       AttName     Rank

1          Color       1

1           type       2

1           kkk        3

2          Color       1

2           type       2     

如上图所示，例如，一些键有 2 个属性，而其他键有 3 个属性，所以当我创建数据透视表然后尝试标记它不起作用的列名时，hpw 我可以更改它吗，正如上面显示的那样，键 1 有 3 个属性名称而 Key 2 只有 2 个导致错误发生

最终数据：

Table2

Family     Assortment Group       Key       Attribute Name      Attribute Value

a                ab                1            Color              Green

a1               ab                1            Color              Yellow

a2               ab                1            type               shirt

a6               ab                1            kkkk               f

a3               ab                2            Color              Red

a4               ab                2            Type               TShirt

a5               ab                2            Color              Yellow

代码

#For loop that loops over key values; key values defined as Zone AG

Finals=[]

Finals2=[]

Finals=pd.DataFrame(Finals)

Finals2=pd.DataFrame(Finals)

for group in Select.groupby('Key'): 

    # group is a tuple where the first value is the Key and the second is the dataframe

    Final2=group[1]

    Family1=Family.merge(Final2, on='Key1', how='inner')

    result=Family1.pivot_table(index=['Family','Assortment Group','Key'], columns='Attribute Name', values='Attribute Value', aggfunc='first').reset_index()

    result.columns=['Family','Assortment Group','Key','Att1','Att2','Att3']

    Finals=Finals.append(result)

追溯

ValueError: Length mismatch: Expected axis has 5 elements, new values have 6 elements

Answer 1

您可以使用列表理解重命名列以生成适量的 Att 列。result.columns = ['Family','Assortment Group','Key']\                  + [f'Att{i}' for i in range(1, result.shape[1]-2)]编辑：解释，result.shape[1]给出结果中的列数。假设它是 5，那么前 3 个是“Family”、“Assortment Group”、“Key”。所以你想要的是再创建 2 个 Att，range(1, result.shape[1]-2)在这种情况下range(1, 3)，[f'Att{i}' for i in range(1, result.shape[1]-2)]然后将迭代 i=1 和 i=2 以创建列表['Att1', 'Att2']。将此列表添加到具有前 3 列名称的列表中以获得正确的列数