Python将列表中的类似子字符串分组

在本教程中，我们将编写一个程序，该程序将列表中的相似子字符串分组。让我们看一个例子来更清楚地理解它。

输入值

strings = ['tutorials-python', 'tutorials-c', 'tutorials-java', 'tutorials-javascript', 'python-1', 'python-2',
'javascript-1']

输出结果

[['tutorials-python', 'tutorials-c', 'tutorials-java', 'tutorials-javascript'], ['python-1', 'python-2'],
['javascript-1']]

我们将使用itertools模块中的groupby方法来解决该问题。该GROUPBY方法组中的所有类似的字符串转换为ITER对象。对于给定的列表，我们用-分割了字符串，并将字符串的第一部分传递给groupby方法。

让我们看看解决这个问题所涉及的步骤。

• 初始化字符串列表。

• 导入itertools模块。

• 初始化一个空列表。

• 现在，将字符串和一个lambda函数传递给itertools.groupby方法。

• lambda函数在使用-拆分后应返回字符串的第一部分

• 该GROUPBY方法将返回元素及其集团的元组的列表。

• 在每次迭代中，将一组相似的元素转换为一个列表。

• 将列表追加到空列表。

• 打印结果。

示例

# importing the itertools module
import itertools
# initializing the strings
strings = ['tutorials-python', 'tutorials-c', 'tutorials-java', 'tutorials-javascript', 'python-1', 'python-2', 'javascript-1']
# empty list
result = []
# iterator
# lambda function will return first part from the string
iterator = itertools.groupby(strings, lambda string: string.split('-')[0])
# iterating over the result
# element and its group
for element, group in iterator:
   # appending the group by converting it into a list
   result.append(list(group))
# printing the result
print(result)

输出结果

如果运行上面的代码，则将得到以下结果。

[['tutorials-python', 'tutorials-c', 'tutorials-java', 'tutorials-javascript'],
ython-1', 'python-2'], ['javascript-1']]

结论

如果您对本教程有任何疑问，请在评论部分中提及。