pandas 以复制品为枢轴
问题描述
示例数据如下:
d = pd.DataFrame({'name': ['Adam', 'Adam', 'Bob', 'Bob', 'Craig'],
'number': [111, 222, 333, 444, 555],
'type': ['phone', 'fax', 'phone', 'phone', 'fax']})
name number type
------ ------ -----
Adam 111 phone
Adam 222 fax
Bob 333 phone
Bob 444 phone
Craig 555 fax
我正在尝试将号码(电话和传真)转换为宽格式,理想输出:
name fax phone
---- ----- -----
Adam 222.0 111.0
Bob NaN 333.0
Bob NaN 444.0
Craig 555.0 NaN
当我尝试使用pivot
方法并运行以下代码时:
p = d.pivot(index='name', columns = 'type', values='number').reset_index()
我收到错误ValueError: Index contains duplicate entries, cannot reshape
,因为Bob有两个电话号码。
是否有解决此问题的方法?
解决方案
这里,您可以使用cumcount
创建附加密钥
d['key']=d.groupby(['name','type']).cumcount()
p = d.pivot_table(index=['key','name'], columns = 'type', values='number',aggfunc='sum').reset_index()
p
Out[71]:
type key name fax phone
0 0 Adam 222.0 111.0
1 0 Bob NaN 333.0
2 0 Craig 555.0 NaN
3 1 Bob NaN 444.0
相关文章