不依赖Pandas,如何快速分组二维列表?

ID:21648 / 打印

不依赖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 依次对数据进行分组,并通过列表推导式将每组提取出来,从而有效地减少了内存占用并提高了效率。

上一篇: 如何高效读取NumPy ndarray中的数据?
下一篇: Anaconda的Channel是什么?如何添加、管理和启用/禁用它?

作者:admin @ 24资源网   2025-01-14

本站所有软件、源码、文章均有网友提供,如有侵权联系308410122@qq.com

与本文相关文章

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。