数据分组统计

缘起

在做业务报表时需要统计出如下图的结果。

即 横坐标是10节课以内的观看人数是244人， 10-20 节课的人数是233人，依次内推。

实现

模拟数据结构如下

这里我首先自己去实现了这个，代码如下：

自实现一

        result_dict = {}
        datum = 5
        for order in order_list:
            res_div = (order['amount'] - 1) // datum

            key = str((res_div+1) * datum)
            if key not in result_dict:
                result_dict.update({key: 1})
            else:
                result_dict.update({key: result_dict.get(key) + 1})
        apply_data = list(result_dict.items())

使用第三方库

        from itertools import groupby
        
        result_dict = {}
        datum = 5
        for k, g in groupby(order_list, key=lambda x: (x['amount'] - 1) // datum):
            result_dict['{}'.format((k+1) * datum)] = int(len(list(g)))

        apply_data = list(result_dict.items())

的确，熟练掌握第三方库会让程序看起来更简短，但是不能忘了自己也要能实现