不依赖Pandas，如何快速分组二维列表？

大家好，今天本人给大家带来文章《不依赖Pandas，如何快速分组二维列表？》，文中内容主要涉及到，如果你对文章方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

二维列表如何快速分组，不依赖 pandas

在不使用 pandas 的情况下，快速对二维列表进行分组是一个常见问题。现在让我们探讨一种有效的方法：

为了按照组名对列表进行分组并保持顺序，我们可以使用自定义函数：

def group_name_fun(lst):
    last = ''
    arr = []
    for ii in range(len(lst)):
        if not lst[ii][0] == last:
            arr.append(ii)
            last = lst[ii][0]
    arr.append(len(lst))
    lst = [lst[arr[ii - 1]:arr[ii]] for ii in range(1, len(arr))]
    return lst

对于大数组，出于效率考虑，建议采用另一种方法：

import itertools

def group_large_array(data):
    grouped_data = {}
    for key, group in itertools.groupby(data, lambda x: x[0]):
        grouped_data[key] = [item[1] for item in group]

    # 转换为列表形式
    result = [grouped_data[key] for key in grouped_data]
    return result

该方法使用 itertools.groupby 依次对数据进行分组，并通过列表推导式将每组提取出来，从而有效地减少了内存占用并提高了效率。

本篇关于《不依赖Pandas，如何快速分组二维列表？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于文章的相关知识，请关注golang学习网公众号！