如何对长度不一的Python Pandas groupby对象执行切片操作?
解决方案
你可以通过以下两种常用方式实现选中每个分组除最后一条外的所有内容:
- 方法1:使用
apply结合iloc切片(兼容性最好,适配所有Pandas版本)
要获取完整行数据,直接执行:
result = grouped_by_Set.apply(lambda x: x.iloc[:-1]).reset_index(drop=True, level=0)
如果仅需要获取Value列的数据,可以简化为:
result_value = grouped_by_Set['Value'].apply(lambda x: x.iloc[:-1])
- 方法2:使用
groupby.head方法(适配Pandas 1.1.0及以上版本)head支持传入可调用对象动态计算每个分组要取的行数:
result = grouped_by_Set.head(lambda x: len(x)-1)
原方案失效原因
nth方法不支持Python原生的切片语法,仅支持传入单个整数、整数列表来指定要提取的行位置,因此0:-1的写法无法被识别。
内容的提问来源于stack exchange,提问作者H. Vabri
相关产品推荐
相关产品推荐

