gpt4 book ai didi

python - 反转 pandas 中的 get_dummies 编码

转载 作者:太空狗 更新时间:2023-10-29 21:33:17 24 4
gpt4 key购买 nike

列名是:ID,1,2,3,4,5,6,7,8,9。

col 值为 0 或 1

我的数据框是这样的:

 ID     1    2    3    4    5    6   7   8   9 

1002 0 1 0 1 0 0 0 0 0
1003 0 0 0 0 0 0 0 0 0
1004 1 1 0 0 0 0 0 0 0
1005 0 0 0 0 1 0 0 0 0
1006 0 0 0 0 0 1 0 0 0
1007 1 0 1 0 0 0 0 0 0
1000 0 0 0 0 0 0 0 0 0
1009 0 0 1 0 0 0 1 0 0

我想要ID前面的列名,其中一行的值为1。

我想要的 Dataframe 应该是这样的:

 ID      Col2
1002 2 // has 1 at Col(2) and Col(4)
1002 4
1004 1 // has 1 at col(1) and col(2)
1004 2
1005 5 // has 1 at col(5)
1006 6 // has 1 at col(6)
1007 1 // has 1 at col(1) and col(3)
1007 3
1009 3 // has 1 at col(3) and col(7)
1009 7

请帮助我,在此先感谢

最佳答案

漂亮的一行:)

new_df = df.idxmax(axis=1)

关于python - 反转 pandas 中的 get_dummies 编码,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50607740/

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