我有这样一个数据帧:
df_1 = pd.DataFrame({
'ID' : ['A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'C', 'C', 'C', 'C', 'C'],
'VAL' : ['shoes', 'flowers', 'chairs', 'apples', 'dice', 'shoes', 'apples',
'curtain', 'sand', 'socks', 'necklacs', 'tables', 'dishes', 'apples'],
'SEQ' : [0, 1, 2, 3, 4, 0, 1, 2, 3, 0, 1, 2, 3, 4]
})
ID VAL SEQ
0 A shoes 0
1 A flowers 1
2 A chairs 2
3 A apples 3
4 A dice 4
5 B shoes 0
6 B apples 1
7 B curtain 2
8 B sand 3
9 C socks 0
10 C necklacs 1
11 C tables 2
12 C dishes 3
13 C apples 4
我想切片到某个值的行,例如,切片到ID
的每个apple
组的所有行:
Out[110]:
ID VAL SEQ
0 A shoes 0
1 A flowers 1
2 A chairs 2
3 A apples 3
4 B shoes 0
5 B apples 1
6 C socks 0
7 C necklacs 1
8 C tables 2
9 C dishes 3
10 C apples 4
idxmax
,groupby
,concat
我正在使用
transform
GroupBy.cumsum
是你的朋友:如果ID可能以您要查找的术语结尾,那么上面的
shift
解决方案(虽然方便)将是不合适的。将GroupBy.apply
与cumsum
一起使用:相关问题 更多 >
编程相关推荐