我有这样一个数据框:
Name | Assignment Group
ABC | 1
DEF | 1
FGH | 1
ABC | 1
IJK | 1
我现在想Assignment Group
根据 column分配一个特定的Name
。如:名称ABC
和IJK
属于组A
和DEF
以及FGH
对B
。
最后我想得到一个这样的数据框:
Name | Assignment Group
ABC | A
DEF | B
FGH | B
ABC | A
IJK | A
我怎么能在python中做到这一点?可以一组一组地做这个。
尝试通过map()
:
d={'A':['ABC','IJK'],'B':['DEF','FGH']}
#created a dict for mapping values
df['Assignment Group']=df['Name'].map({i:k for k,v in d.items() for i in v})
或者
通过replace()
:
df['Assignment Group']=df['Name'].replace({i:k for k,v in d.items() for i in v})
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句