python - pd.get_dummies wrt 特定列

Question

这个问题已经有答案了:

How can I pivot a dataframe? (5 个回答)

已关闭 5 年前。

DF 说明:

Pandas DF 有两列“ID”、“名称”，其中同一“ID”有多个“名称”条目

用例:

IP:

 ID          Names
 1           func_A
 1           tunc_B
 2           rain
 2           fire

所需操作:

一个新的 DF，其条目如下

ID    func_A       func_B   rain fire
 1        1         1         0   0
 2        0         0         1   1

我尝试了两种使用 groupBy 和 get_dummies 的方法，但我没有得到所需的操作。任何有关如何实现这一目标的建议都受到高度赞赏。谢谢。

Answer 1

您可以使用pd.crosstab(df.ID, df.Names)。

输出:

Names  fire  func_A  rain  tunc_B
ID
1         0       1     0       1
2         1       0     1       0