让我们考虑一个包含许多不同长度字符串的列表。在本文中,我们将看到如何将这些元素分为几组,其中每组中的字符串长度相等。
我们设计了一个for循环,该循环将遍历列表的每个元素,并且仅在其长度与现有元素的长度匹配的列表上发生它。
listA = ['Monday','Thursday','Friday','Saturday','Sunday'] # Given list print("Given list : \n",listA) # Categorize by string size len_comp = lambda x, y: len(x) == len(y) res = [] for sub_list in listA: ele = next((i for i in res if len_comp(sub_list, i[0])), []) if ele == []: res.append(ele) ele.append(sub_list) # Result print("The list after creating categories : \n",res)
输出结果
运行上面的代码给我们以下结果-
Given list : ['Monday', 'Thursday', 'Friday', 'Saturday', 'Sunday'] The list after creating categories : [['Monday', 'Friday', 'Sunday'], ['Thursday', 'Saturday']]
在这种方法中,我们首先根据元素的长度拍摄所有元素,然后将其作为itertools模块的一部分按功能进行分组。
from itertools import groupby listA = ['Monday','Thursday','Friday','Saturday','Sunday'] # Given list print("Given list : \n",listA) # Categorize by string size get_len = lambda x: len(x) sub_list = sorted(listA, key = get_len) res = [list(ele) for i, ele in groupby(sub_list, get_len)] # Result print("The list after creating categories : \n",res)
输出结果
运行上面的代码给我们以下结果-
Given list : ['Monday', 'Thursday', 'Friday', 'Saturday', 'Sunday'] The list after creating categories : [['Monday', 'Friday', 'Sunday'], ['Thursday', 'Saturday']]