gpt4 book ai didi

python - 在 DataFrame.groupby 的情况下,如何根据另一列的最大值获取列的值

转载 作者:太空宇宙 更新时间:2023-11-04 09:47:20 26 4
gpt4 key购买 nike

我有一个看起来像这样的数据框。

id YearReleased 艺术家数
168 2015 缪斯 1
169 2015 蕾哈娜 3
170 2015 泰勒· swift 2
171 2016 詹妮弗·洛佩兹 1
172 2016 蕾哈娜 3
173 2016 黑社会 1
174 2017 酷玩乐队 1
175 2017 红发艾德 2

我想获取每年的最大计数,然后获取相应的艺术家姓名。

像这样:

年发行艺术家

2015 蕾哈娜
2016 蕾哈娜
2017 红发艾德

我尝试使用循环遍历数据框的行,并创建另一个字典,键为年份,值为艺术家。但是当我尝试将该字典转换为数据框时,键映射到列而不是行。

有人可以指导我有一个更好的方法来解决这个问题,而不必遍历数据帧,而是使用一些内置的 pandas 方法来实现这个吗?

最佳答案

idxmax

df.loc[df.groupby('YearReleased')['count'].idxmax()]
Out[445]:
id YearReleased Artist count
1 169 2015 Rihanna 3
4 172 2016 Rihanna 3
7 175 2017 EdSheeran 2

关于python - 在 DataFrame.groupby 的情况下,如何根据另一列的最大值获取列的值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49263437/

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