[英]Python (pandas) - How to group values in one column and then delete or keep that group based on values in another column
[英]Pandas group by but keep another column
假设我有一个看起来像这样的数据框
date location year
0 1908-09-17 Fort Myer, Virginia 1908
1 1909-09-07 Juvisy-sur-Orge, France 1909
2 1912-07-12 Atlantic City, New Jersey 1912
3 1913-08-06 Victoria, British Columbia, Canada 1912
我想使用 pandas groupby 函数来创建一个输出,该输出显示每年的事件总数,但同时保留将显示当年位置之一的位置列。 任何一个工作。 所以它看起来像这样:
total location
year
1908 1 Fort Myer, Virginia
1909 1 Juvisy-sur-Orge, France
1912 2 Atlantic City, New Jersey
这可以在不进行时髦加入的情况下完成吗? 我能得到的最远的是使用普通的 groupby
df = df.groupby(['year']).count()
但这只会给我这样的东西
location
year
1908 1 1
1909 1 1
1912 2 2
如何显示此数据框中的位置之一?
您可以使用groupby.agg
并使用'first'
来提取每个组中的第一个位置:
res = df.groupby('year')['location'].agg(['first', 'count'])
print(res)
# first count
# year
# 1908 Fort Myer, Virginia 1
# 1909 Juvisy-sur-Orge, France 1
# 1912 Atlantic City, New Jersey 2
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.