我有一个带有列的 pandas 数据框(除其他外)user_id
and start_time
。我想高效且易读地找到与每个用户的最大值关联的所有行start_time
.
例如,如果这是我的数据:
user_id start_time A B C
1 37 a b c
1 45 d e f
1 45 g h i
2 58 j k l
2 17 m n o
2 58 p q r
3 2 s t u
那么我期望找到
user_id start_time A B C
1 45 d e f
1 45 g h i
2 58 j k l
2 58 p q r
3 2 s t u
我一直在想出一些类似的解决方案pandas DataFrame 中数据的条件选择 https://stackoverflow.com/questions/28111791/conditional-selection-of-data-in-pandas-dataframe,但这会发现user_id
与最新的start time
,而不是选择每个用户最大的表start_time
's.
当然,如果我手动迭代DataFrame,这很容易,但效率很低。
感谢您的指点。
为了方便未来的读者,生成数据帧:
columns = ['user_id', 'start_time', 'A', 'B', 'C']
LoL = [
[1, 37, 'a', 'b', 'c'],
[1, 45, 'd', 'e', 'f'],
[1, 45, 'g', 'h', 'i'],
[2, 58, 'j', 'k', 'l'],
[2, 17, 'm', 'n', 'o'],
[2, 58, 'p', 'q', 'r'],
[3, 2, 's', 't', 'u']]
pd.DataFrame = (LoL, columns=columns)