Question

我有一个这样的熊猫数据集：

  Brand AssociatedWord  Weight
0  pepsi           red      10
1  pepsi        yellow       3
2  coke            red       5
3  coke           grey       5
4  coke           pink       2

我需要将其转换为以下矩阵：

  Brand   red   yellow   grey   pink
0  pepsi   10        3      0      0
1  coke     5        0      5      2

现在，每一行都是一个品牌，并且每个相关单词都有一列，其中报告了关联的权重。零值表示缺少关联。列的顺序并不重要。你能帮我吗？

Answer 1

使用DataFrame.pivot_table：

new_df=df.pivot_table(index='Brand',columns='AssociatedWord',values='Weight',fill_value=0).reset_index()
print(new_df)

AssociatedWord  Brand  grey  pink  red  yellow
0                coke     5     2    5       0
1               pepsi     0     0   10       3

注意：AssociatedWord是列的名称，您可以使用以下方式进行更改：

new_df.columns.name=None

   Brand  grey  pink  red  yellow
0   coke     5     2    5       0
1  pepsi     0     0   10       3

您还可以使用set_index + unstack：

new_df=df.set_index(['Brand','AssociatedWord']).unstack(fill_value=0).reset_index()
print(new_df)


new_name        Brand Weight                
AssociatedWord          grey pink red yellow
0                coke      5    2   5      0
1               pepsi      0    0  10      3

熊猫在文档单词矩阵中转换文档单词列表

1 个答案: