使用Python中的pandas库的groupby方法和apply方法,结合lambda表达式进行筛选,示例代码如下:
import pandas as pd
# 创建数据集
data = pd.DataFrame({'group': ['A', 'A', 'B', 'B', 'B', 'C', 'C'], 'value': [1, 2, 3, 4, 5, 6, 7]})
# 对group进行分组,根据组别实现不同行数的筛选
filtered_data = data.groupby('group').apply(lambda x: x[:2] if x['group'].iloc[0] == 'A' else x[:3] if x['group'].iloc[0] == 'B' else x[:1])
print(filtered_data)
以上代码中,我们首先创建了一个包含分组数据的DataFrame。然后,使用groupby方法将数据集按组进行分组。在apply方法中,我们使用lambda表达式根据组别进行不同数目的筛选:对于组A,选取前两行;对于组B,选取前三行;对于其它组,选取前一行。最后,输出过滤后的DataFrame。
上一篇:Anzure搜索摄取事件