在Python中,可以使用pandas
库来按组应用不同的筛选条件。
首先,导入pandas
库:
import pandas as pd
接下来,创建一个包含数据的数据帧(DataFrame):
data = {'Group': ['A', 'A', 'B', 'B', 'B', 'C', 'C'],
'Value': [1, 2, 3, 4, 5, 6, 7]}
df = pd.DataFrame(data)
现在,我们可以定义一个函数来根据组应用不同的筛选条件:
def filter_func(group):
if group['Group'].iloc[0] == 'A':
return group[group['Value'] > 1]
elif group['Group'].iloc[0] == 'B':
return group[group['Value'] > 3]
elif group['Group'].iloc[0] == 'C':
return group[group['Value'] > 5]
else:
return group
然后,我们可以使用groupby
方法来按组应用这个函数:
result = df.groupby('Group').apply(filter_func)
最后,打印结果:
print(result)
这将按组应用不同的筛选条件,并打印出结果数据帧。
请注意,上述代码示例中的筛选条件只是示例,你可以根据实际需求自定义不同的筛选条件。