Python按字符串大小对给定列表进行分类

让我们考虑一个包含许多不同长度字符串的列表。在本文中,我们将看到如何将这些元素分为几组,其中每组中的字符串长度相等。

带for循环

我们设计了一个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']]