python - Python将map与groupby和transform结合起来

Question

我在 Python 中组合几个概念时遇到了麻烦；groupby、映射和变换。我有一个数据框，我希望通过基于组转换现有列来创建新列。像这样：

s = df[df['type'].eq('office')].groupby(['user','date']).transform('any')
df.loc[:,'type2'] = df['type'].s.map({True:'office',False:'remote'})

所以我的数据框看起来像这样

user     date    type   type2
ron     12/1/19  office  office
ron     12/1/19  remote  office
april   12/1/19  office  office
leslie  12/1/19  remote  office
leslie  12/1/19  office  office
leslie  2/1/20   office  office

但我收到以下错误：

AttributeError: 'Series' object has no attribute 's'

我以为我正确设置了它，但无法让它工作。感谢指导 谢谢

标签： pythonpandaspandas-groupbytransform

Answer 1

修复你的代码

s = df['type'].eq('office').groupby([df['user'],df['date']]).transform('any')
df['type2'] = s.map({True:'office',False:'remote'})