python - Pandas 在文档词矩阵中转换文档词列表

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

  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

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

最佳答案

使用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

关于python - Pandas 在文档词矩阵中转换文档词列表，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/58354692/