将数据按照ID分组,然后应用条件对选定的列进行过滤。这可以使用 pandas 库中的 groupby 和 loc 方法实现。
示例代码如下:
import pandas as pd
# 创建示例数据
data = {
'id': [1, 2, 3, 1, 2, 3],
'value': [10, 20, 30, 15, 25, 35]
}
df = pd.DataFrame(data)
# 按 ID 分组并过滤列 "value" 的值
filtered_df = df.groupby('id').apply(lambda x: x.loc[x['value'] > 20])
print(filtered_df)
输出:
id value
id
2 2 25
3 3 30
3 3 35
上一篇:按id分组并打印结果
下一篇:按ID分组并获取多项式分布列