我有如下所示的数据格式:
import pandas as pd
data={'location':['LA','LA','SF','LA','SF','LA','SF'], 'col2': [1, 3, 5, 2, 8, 5, 7], 'col3': [11, 14, 54, 67, 2, 9, 45]}
df=pd.DataFrame(data,columns=['location', 'col2', 'col3'])
print df
location col2 col3
0 LA 1 11
1 LA 3 14
2 SF 5 54
3 LA 2 67
4 SF 8 2
5 LA 5 9
6 SF 7 45
我想按“位置”进行分组,然后取我所分组的所有行的"col2“的最大值,但我不想要"col3”的最大值。我希望"col3“的值与”col3“行的值相同,该行对于该位置有最大的"col2”。
location col2 col3
0 LA 5 9
1 SF 8 2
谢谢,如果有什么不清楚的话,请告诉我。
发布于 2017-01-06 20:50:19
groupby('location').col2
如预期idxmax()
并在完全df
上的loc
中使用这个df.loc[df.groupby('location').col2.idxmax()]
location col2 col3
5 LA 5 9
4 SF 8 2
https://stackoverflow.com/questions/41514097
复制相似问题