Python 按另一个列表对子列表进行分组

在Python中，我们可以使用各种方法将子列表按另一个列表进行分组，例如使用字典和itertools.groupby()函数，使用嵌套列表推导式等。将子列表按另一个列表进行分组在分析大型数据集和数据分类时非常有用。它还用于文本分析和自然语言处理。在本文中，我们将探讨在Python中按另一个列表对子列表进行分组的不同方法，并了解它们的实现。

方法1：使用字典

可以非常直接地使用字典来按另一个列表对子列表进行分组。让我们通过一个示例来了解使用字典将子列表按另一个列表进行分组的用法。

语法

list_name.append(element)

在这里，元素是要添加到列表末尾的元素。 追加方法将此元素放置在列表的末尾。

示例

在以下示例中，我们定义了一个名为group_sublists的函数，它接受两个参数：sublists（子列表的列表）和grouping_list（确定分组顺序的列表）。在函数内部，我们创建一个空字典groups来存储按键分组的子列表。我们遍历sublists列表中的每个子列表。假设每个子列表的第一个元素是键，我们提取它并检查它是否存在于groups字典中。如果存在，则将当前子列表追加到该键的现有子列表列表中。否则，在groups字典中创建一个新的键-值对，其中键是该键，值是当前子列表。最后，我们返回一个列表推导式，按照grouping_list指定的顺序检索分组的子列表。

def group_sublists(sublists, grouping_list):
    groups = {}

    for sublist in sublists:
        key = sublist[0]  # Assuming the first element of each sublist is the key
        if key in groups:
            groups[key].append(sublist)
        else:
            groups[key] = [sublist]

    return [groups[key] for key in grouping_list]

# Example usage
sublists = [[1, 'apple'], [2, 'banana'], [1, 'orange'], [2, 'grape']]
grouping_list = [1, 2]

result = group_sublists(sublists, grouping_list)
print(result)

输出

[[[1, 'apple'], [1, 'orange']], [[2, 'banana'], [2, 'grape']]]

方法2：使用itertools.groupby()函数

Python的itertools模块提供了一个方便的函数称为groupby()，它可以根据键函数将元素分组。让我们通过一个示例来理解这个函数。

语法

list_name.append(element)

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

itertools.groupby(iterable, key=None)

这里，iterable是输入的可迭代对象，可以是任何你想要进行分组的序列， key=None 是一个可选参数，可以是一个用作分组键的函数。如果没有提供key函数，就会使用元素本身作为分组的键。

示例

在下面的示例中，我们首先根据key（假设为第一个元素）对子列表进行排序。然后，我们创建一个空列表result来存储分组后的子列表。接下来，我们遍历由itertools.groupby()生成的分组。groupby()函数接受两个参数：可迭代对象（在本例中为子列表）和用于提取每个子列表键的key函数（一个lambda函数）。它返回由键和包含分组后子列表的迭代器组成的对。在循环内部，我们检查分组列表中是否存在键。如果存在，我们使用list(group)将迭代器转换为列表，并将其附加到result列表中。最后，我们返回包含分组后子列表的result列表。

import itertools

def group_sublists(sublists, grouping_list):
    sublists.sort(key=lambda x: x[0])  # Sort the sublists based on the key

    result = []
    for key, group in itertools.groupby(sublists, lambda x: x[0]):
        if key in grouping_list:
            result.append(list(group))

    return result

# Example usage
sublists = [[1, 'apple'], [2, 'banana'], [1, 'orange'], [2, 'grape']]
grouping_list = [1, 2]

result = group_sublists(sublists, grouping_list)
print(result)

输出

[[[1, 'apple'], [1, 'orange']], [[2, 'banana'], [2, 'grape']]]

方法3：使用嵌套列表推导

我们可以使用Python编写嵌套的列表推导，用于将子列表按照另一个列表进行分组。让我们看一个示例来看如何实现这一点。

语法

[expression for item in list if condition]

在这里，语法由方括号括起来的表达式和紧随其后的一个for循环组成，该循环在列表上进行迭代。也可以在表达式的末尾添加任何if条件来过滤掉某些元素。

示例

在下面的示例中，我们定义了名为 group_sublists 的函数，它以子列表和分组列表作为参数。我们使用嵌套的列表推导来迭代分组列表中的每个键。对于每个键，我们在子列表上进行迭代，并筛选出只有匹配键（假设为第一个元素）的子列表。然后将这些筛选后的子列表收集到一个新的列表中，表示该键的分组子列表。结果是一个包含分组子列表的列表，每个子列表都包含特定键的分组子列表。

def group_sublists(sublists, grouping_list):
    return [
        [sublist for sublist in sublists if sublist[0] == key]
        for key in grouping_list
    ]

# Example usage
sublists = [[1, 'apple'], [2, 'banana'], [1, 'orange'], [2, 'grape']]
grouping_list = [1, 2]

result = group_sublists(sublists, grouping_list)
print(result)

输出

[[[1, 'apple'], [1, 'orange']], [[2, 'banana'], [2, 'grape']]]

结论

在本文中，我们讨论了如何在Python中按照另一个列表将子列表分组。我们讨论了三种方法：使用字典和利用itertools.groupby()函数，以及使用嵌套的列表推导式。每种方法都有其优点，并且可能根据您程序的具体要求而更适合使用。