我正在根据给定列中值的范围来聚合数据框的内容。我的df
如下所示:
min max names
1 5 ['a','b']
0 5 ['d']
6 8 ['a','c']
3 4 ['e','a']
预期输出为
min=0
和max=5
,获取汇总值,因此名称值将为['a','b','d','e','a']
min=5
和max=10
,获取汇总值,名称值将为['a','d']
感谢您的帮助。
答案 0 :(得分:4)
最直观的方法是过滤,然后聚合。为了解决您的特定问题,我会这样做:
>> df = pd.DataFrame({"min": [1, 0, 6, 3],
"max": [5, 5, 8, 4],
"value": [['a','b'], ['d'], ['a','c'], ['e','a']]})
>> print(df)
min max value
0 1 5 [a, b]
1 0 5 [d]
2 6 8 [a, c]
3 3 4 [e, a]
>> sum_filtered_values = df[(df["max"]<=5) & (df["min"]>=0)].value.sum()
>> print(sum_filtered_values)
['a', 'b', 'd', 'e', 'a']
>> sum_filtered_values = df[(df["max"]<=10) & (df["min"]>=5)].value.sum()
>> print(sum_filtered_values)
['a', 'c']