使用 Python 根据字符串首字母分组列表

在 Python 中，我们可以使用多种方法根据字符串的首字母对列表进行分组，例如使用字典、使用 itertools.groupby、使用 Defaultdict 等。这在各种场景中都很有用，例如组织姓名或对数据进行分类。在本文中，我们将探讨使用 Python 根据字符串的首字母对列表进行分组的不同方法。

方法 1：使用字典

在这种方法中，字典的键将表示首字母，相应的值将是包含以该字符开头的所有字符串的列表。

语法

list_name.append(element)

这里，append() 函数是列表方法，用于将元素添加到列表末尾。它通过将指定元素作为新项添加到列表中来修改原始列表。

示例

在下面的示例中，函数 group_list_by_first_character 以字符串列表作为输入并返回一个字典。它遍历输入列表中的每个字符串并提取首字母。如果首字母已存在于字典中作为键，则将字符串附加到相应的值列表中。否则，将创建一个新的键值对，其中首字母作为键，字符串作为值列表中的第一项。

Open Compiler
def group_list_by_first_character(strings):
    grouped_dict = {}
    for string in strings:
        first_character = string[0]
        if first_character in grouped_dict:
            grouped_dict[first_character].append(string)
        else:
            grouped_dict[first_character] = [string]
    return grouped_dict

# Example usage
strings = ["apple", "banana", "cat", "dog", "elephant"]
grouped_dict = group_list_by_first_character(strings)
print(grouped_dict)

输出

{'a': ['apple'], 'b': ['banana'], 'c': ['cat'], 'd': ['dog'], 'e': ['elephant']}

方法 2：使用 itertools.groupby

Python 的 itertools.groupby 是一个有用的函数，可用于根据特定条件对元素进行分组。要根据首字母对字符串列表进行分组，我们可以使用 itertools.groupby 以及 lambda 函数。

语法

list_name.append(element)

这里，append() 函数是列表方法，用于将元素添加到列表末尾。它通过将指定元素作为新项添加到列表中来修改原始列表。

itertools.groupby(iterable, key=None)

这里，groupby() 方法以可迭代对象作为输入，并带有一个可选的键函数。它返回一个迭代器，该迭代器生成包含可迭代对象中连续键和组的元组。键函数用于确定分组标准。

示例

在下面的示例中，我们导入 itertools 模块并使用 groupby 函数。在应用 groupby 之前，我们使用 sort() 方法对输入列表进行排序。此步骤是必要的，因为 groupby 对具有相同键的连续元素进行操作。通过对列表进行排序，我们确保具有相同首字母的字符串彼此相邻。

lambda 函数 lambda x: x[0] 指定分组标准是每个字符串的首字母。然后，groupby 函数返回一个迭代器，其中包含具有键和表示组的可迭代对象的元组。我们将每个组转换为列表并将其附加到 grouped_list 中。

Open Compiler
import itertools

def group_list_by_first_character(strings):
    strings.sort()  # Sorting the list is necessary for groupby to work correctly
    grouped_list = []
    for key, group in itertools.groupby(strings, lambda x: x[0]):
        grouped_list.append(list(group))
    return grouped_list

# Example usage
strings = ["apple", "banana", "cat", "dog", "elephant"]
grouped_list = group_list_by_first_character(strings)
print(grouped_list)

输出

[['apple'], ['banana'], ['cat'], ['dog'], ['elephant']]

Learn Python in-depth with real-world projects through our Python certification course. Enroll and become a certified expert to boost your career.

方法 3：使用 DefaultDict

另一种根据字符串的首字母对列表进行分组的有用方法是利用 collections 模块中的 defaultdict 类。Defaultdict 是内置 dict 类的子类，它会自动使用默认值初始化缺失的键。

语法

groups[item].append(item)

这里，语法使用 collections 模块中的 defaultdict() 函数初始化一个名为 groups 的 defaultdict 对象，其默认值为一个空列表。代码的第二行使用键 (item) 访问 groups 字典中与此键关联的列表，并将 item 附加到列表中。

示例

在下面的示例中，我们从 collections 模块导入 defaultdict 类。我们创建一个 defaultdict 对象 grouped_dict，其默认值设置为一个空列表。当我们遍历输入列表中的每个字符串时，我们使用首字母作为键并将字符串附加到 grouped_dict 中相应的值列表中。

Open Compiler
from collections import defaultdict

def group_list_by_first_character(strings):
    grouped_dict = defaultdict(list)
    for string in strings:
        grouped_dict[string[0]].append(string)
    return grouped_dict

# Example usage
strings = ["apple", "banana", "cat", "dog", "elephant"]
grouped_dict = group_list_by_first_character(strings)
print(grouped_dict)

输出

defaultdict(<class 'list'>, {'a': ['apple'], 'b': ['banana'], 'c': ['cat'], 'd': ['dog'], 'e': ['elephant']})

结论

在本文中，我们讨论了如何在 Python 中使用不同的方法根据字符串的首字母对列表进行分组。我们使用了字典、itertools.groupby 和 defaultdict 来实现所需的分组。每种方法都有其自身的优势，可能适合于您的应用程序的特定需求。

更新于： 2023年7月17日

499 次浏览

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告