Pandas groupby并从每个正负类型中选择前5个和后5个值

import pandas as pd

# 创建一个示例数据集
data = {'value': [1, -2, 3, -4, 5, -6, 7, -8, 9, -10, 11, -12],
        'type': ['positive', 'negative', 'positive', 'negative', 'positive', 'negative',
                 'positive', 'negative', 'positive', 'negative', 'positive', 'negative']}
df = pd.DataFrame(data)

# 根据正负类型进行分组，并选择前5个和后5个值
result = df.groupby('type').apply(lambda x: x.nsmallest(5, 'value').append(x.nlargest(5, 'value')))

# 打印结果
print(result)