在pandas中获取每个类别的前n个值，保留所有列

import pandas as pd

# 假设有一个DataFrame，其中包含了类别和对应的数值
df = pd.DataFrame({
    'Category': ['A', 'A', 'A', 'B', 'B', 'C'],
    'Value': [1, 2, 3, 4, 5, 6],
    'OtherColumn': ['foo', 'bar', 'baz', 'qux', 'quux', 'corge']
})

# 获取每个类别的前两个值，并保留所有列
n = 2
result = df.groupby('Category').head(n)

print(result)

  Category  Value OtherColumn
0        A      1         foo
1        A      2         bar
3        B      4         qux
4        B      5        quux