按组显示前N个和后N个值的Pandas新数据框架
创始人
2024-09-03 05:30:19
0

下面是一个示例代码,展示了如何使用Pandas创建一个新的数据框架,按组显示前N个和后N个值。

首先,我们导入必要的库:

import pandas as pd

接下来,创建一个示例数据框架:

data = {'Group': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
        'Value': [1, 2, 3, 4, 5, 6, 7, 8, 9]}
df = pd.DataFrame(data)

数据框架如下所示:

  Group  Value
0     A      1
1     A      2
2     A      3
3     B      4
4     B      5
5     B      6
6     C      7
7     C      8
8     C      9

现在,我们可以使用groupby函数按组对数据框架进行分组,并使用apply函数在每个组上应用自定义函数。在自定义函数中,我们可以使用headtail函数选择前N个和后N个值,并使用concat函数将它们连接成一个新的数据框架。

下面是完整的代码示例:

import pandas as pd

data = {'Group': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
        'Value': [1, 2, 3, 4, 5, 6, 7, 8, 9]}
df = pd.DataFrame(data)

def get_top_bottom_n(group, n):
    top_n = group.head(n)
    bottom_n = group.tail(n)
    return pd.concat([top_n, bottom_n])

N = 2
new_df = df.groupby('Group').apply(get_top_bottom_n, n=N).reset_index(drop=True)

新的数据框架new_df如下所示:

  Group  Value
0     A      1
1     A      2
2     A      2
3     A      3
4     B      4
5     B      5
6     B      5
7     B      6
8     C      7
9     C      8
10    C      8
11    C      9

在这个示例中,我们选择了每个组的前2个和后2个值,所以新的数据框架中有12个行。

相关内容

热门资讯

七分钟辅助!丽水茶苑苹果手机辅... 七分钟辅助!丽水茶苑苹果手机辅助,本来是真的有辅助教程(有挂方式)1、实时丽水茶苑苹果手机辅助透视辅...
第一分钟辅助!闲来辅助神器下载... 第一分钟辅助!闲来辅助神器下载2022,好像真的有辅助方法(有挂教程)1、不需要AI权限,帮助你快速...
九分钟辅助!丽水都莱辅助工具试... 九分钟辅助!丽水都莱辅助工具试用,确实存在有辅助神器(有挂方法)九分钟辅助!丽水都莱辅助工具试用,确...
第一分钟辅助!蛮王辅助器,好像... 第一分钟辅助!蛮王辅助器,好像是有辅助方法(有挂教学)1、首先打开蛮王辅助器辅助器下载最新版本,在蛮...
第六分钟辅助!潮汕汇挂,一贯真... 第六分钟辅助!潮汕汇挂,一贯真的是有辅助插件(有挂辅助)1、这是跨平台的潮汕汇挂轻量版有透视,在线的...
六分钟辅助!微信开心泉州辅助器... 六分钟辅助!微信开心泉州辅助器,一直有辅助器(有挂教学)1、下载好微信开心泉州辅助器透视辅助下载之后...
第3分钟辅助!佛手十三道破解版... 第3分钟辅助!佛手十三道破解版安卓,竟然真的有辅助攻略(有挂存在)1、让任何用户在无需佛手十三道破解...
2分钟辅助!sohoo竞技联盟... 2分钟辅助!sohoo竞技联盟辅助,切实真的有辅助脚本(有挂技术)1.sohoo竞技联盟辅助 选牌创...
第8分钟辅助!心悦手游辅助器,... 第8分钟辅助!心悦手游辅助器,原来真的是有辅助技巧(确实有挂);1、每一步都需要思考,不同水平的挑战...
第十分钟辅助!广东雀神祈福真的... 第十分钟辅助!广东雀神祈福真的有用吗,都是是有辅助技巧(有挂方略)1、下载好广东雀神祈福真的有用吗透...