Python - 按列表中的第 K 列对记录进行分组

在Python中，我们经常需要对列表中的记录进行分组。如果我们想按照列表中的第 K 列对记录进行分组，可以使用一些内置的方法和函数来实现这个目标。

方法一：使用itertools.groupby函数

itertools.groupby函数是一个非常有用的函数，可以根据指定的键对列表中的元素进行分组。在这种情况下，我们可以使用operator.itemgetter函数作为键，来选择要分组的列。

import itertools
import operator

def group_by_column(records, k):
    # 按照第 k 列进行排序
    records.sort(key=operator.itemgetter(k))

    # 利用itertools.groupby函数对records进行分组
    groups = []
    for key, group in itertools.groupby(records, key=operator.itemgetter(k)):
        groups.append(list(group))

    return groups

这里的records是包含记录的列表，k是要分组的列的索引。函数首先对列表进行排序，然后使用itertools.groupby函数对排序后的列表进行分组。最后，将分组结果存储在groups列表中并返回。

方法二：使用collections.defaultdict进行分组

collections.defaultdict是一个特殊的字典，它允许我们为不存在的键提供一个默认值。我们可以利用这个特性来创建一个以第 K 列值作为键的字典，并将记录添加到相应的键中。

from collections import defaultdict

def group_by_column(records, k):
    groups = defaultdict(list)

    for record in records:
        groups[record[k]].append(record)

    return list(groups.values())

在这个方法中，我们首先创建一个defaultdict对象groups，默认值为一个空列表。然后，我们遍历records列表，将每个记录添加到以第 K 列值为键的列表中。最后，我们将分组的结果转换为列表并返回。

示例用法

records = [
    [1, 'Apple', 5],
    [2, 'Banana', 3],
    [3, 'Orange', 5],
    [4, 'Apple', 2],
    [5, 'Banana', 1],
    [6, 'Orange', 4]
]

# 按照第二列进行分组
groups = group_by_column(records, 1)

for group in groups:
    print(group)

这个示例中，我们有一个包含记录的列表records。我们使用group_by_column函数按照第二列对记录进行分组，并打印出分组结果。

输出结果为：

[[1, 'Apple', 5], [4, 'Apple', 2]]
[[2, 'Banana', 3], [5, 'Banana', 1]]
[[3, 'Orange', 5], [6, 'Orange', 4]]

以上是一种按照列表中的第 K 列对记录进行分组的方法。无论是使用itertools.groupby函数还是collections.defaultdict，都可以根据具体需求选择最适合的方法进行分组。希望这篇介绍对你有帮助！