如何从熊猫中的一组数据中获取行的子集？
如何从熊猫中的一组数据中获取行的子集？
提问于 2018-11-06 13:20:30
我有一个带有列ID和二进制列的dataframe，如下所示
ID BINARY_MASK
0 101 1
1 101 0
2 101 1
3 101 1
4 101 1
5 101 1
6 101 0
7 101 1
8 102 1
9 102 1
11 102 1
12 102 1
13 102 0
14 102 0我要做的是获取前四个连续条目，即每个ID组1。我希望看到的结果如下：
ID BINARY_MASK
2 101 1
3 101 1
4 101 1
5 101 1
8 102 1
9 102 1
11 102 1
12 102 1组内有四个连续索引的组内的索引每个组不同，如示例中所示。我该怎么做？
我已经尝试过提供给by Bill G in this question的解决方案，但这对我没有用。
使用Pandas数据仓库和Python3.6
Stack Overflow用户
发布于 2018-11-06 13:22:06
使用groupby + head：
df[df['BINARY_MASK']==1].groupby('ID').head(4)
ID BINARY_MASK
0 101 1
2 101 1
3 101 1
4 101 1
8 102 1
9 102 1
11 102 1
12 102 1页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/53172751
复制相关文章
相似问题
腾讯云开发者
