我有一个包含玩家、队伍和胜利的DataFrame。我想创建一个新的DataFrame,其中队伍作为索引,并且表示玩家x是否在该队伍中,以及该队伍是否获胜。
pd.DataFrame(data=[['Team A', 1], ['Team B', 0], ['Team B', 0], ['Team A', 1]], columns=['TEAM', 'WIN'], index=['Player 1', 'Player 2', 'Player 3', 'Player 4']) TEAM WINPlayer 1 Team A 1Player 2 Team B 0Player 3 Team B 0Player 4 Team A 1
期望的结果:
pd.DataFrame(data=[[1, 1, 0, 0, 1], [0, 0, 1, 1, 0]], columns=['Player 1', 'Player 2', 'Player 3', 'Player 4', 'WIN'], index=['Team A', 'Team B']) Player 1 Player 2 Player 3 Player 4 WINTeam A 1 1 0 0 1Team B 0 0 1 1 0
回答:
如果我理解正确的话,你可以使用pivot_table()
方法:
In [96]: df.reset_index().pivot_table(index='TEAM', columns='index', values='WIN', fill_value=0)Out[96]:index Player 1 Player 2 Player 3 Player 4TEAMTeam A 1 0 0 1Team B 0 0 0 0