在给定的数据中,保留每个ID的特定行数可以通过以下代码示例实现:
import pandas as pd
# 创建示例数据
data = {'ID': [1, 1, 1, 2, 2, 2, 3, 3, 3],
'Value': [10, 20, 30, 40, 50, 60, 70, 80, 90]}
df = pd.DataFrame(data)
# 定义每个ID要保留的行数
rows_to_keep = 2
# 通过分组和head方法保留每个ID的特定行数
df_filtered = df.groupby('ID').head(rows_to_keep)
# 输出结果
print(df_filtered)
输出结果为:
ID Value
0 1 10
1 1 20
3 2 40
4 2 50
6 3 70
7 3 80
在上述代码中,首先创建了一个包含ID和Value两列的示例数据DataFrame。然后,定义了每个ID要保留的行数为2。接下来,使用groupby
方法按ID进行分组,并通过head
方法保留每个组的前两行。最后,输出结果显示了保留特定行数后的DataFrame。
上一篇:保留每个id的前两个日期
下一篇:保留每个键的最小值的Java映射