根据Pandas中另一列的值范围汇总一列的内容

Question

我正在根据给定列中值的范围来聚合数据框的内容。我的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']

感谢您的帮助。

Answer 1

最直观的方法是过滤，然后聚合。为了解决您的特定问题，我会这样做：

>> 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']