gpt4 book ai didi

python - 根据现有列中的值创建新列并合并行

转载 作者:行者123 更新时间:2023-12-04 10:39:24 25 4
gpt4 key购买 nike

我的数据框包含一列 ID 和一列主题。最终,我想按主题对我的数据进行聚类,以便我可以看到按主题聚类的 ID 组。

为了做到这一点,我希望将我的数据框作为每个 ID 的一行结束,每个主题的一列包含 1 或 0 值,以指示该主题是否被该行的 ID 占用。

输入示例:

ID      Subject
John History
John English
Jane Maths
Jane Science
Jane History
Mark History
Mark Science
Mark Mathematics

预期输出:
Subject  English  History  Mathematics  Maths  Science
ID
Jane 0 1 0 1 1
John 1 1 0 0 0
Mark 0 1 1 0 1

最佳答案

您可以使用 pivot_table() 执行以下操作

 df.pivot_table(index='ID', columns='Subject', aggfunc=len,fill_value=0)

关于python - 根据现有列中的值创建新列并合并行,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60005602/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com