博客

Python 键值分组或分区数据

  • Post category:Python

Python 键值分组或分区数据，一般指将数据按照某个规则进行分类，并按照分类结果进行分组或分区。Python中有很多用于键值分组或分区数据的函数和库，比如使用Python标准库中的collections库中的defaultdict函数实现简单的键值对分组，也可以使用pandas库中的groupby函数实现更加复杂的分组。

下面是一个实现键值分组的简单示例代码：

from collections import defaultdict

data = [
    ('color', 'red'),
    ('fruit', 'apple'),
    ('color', 'blue'),
    ('animal', 'elephant'),
    ('fruit', 'orange'),
    ('animal', 'lion'),
]

# 使用defaultdict函数实现分组
grouped_data = defaultdict(list)
for key, value in data:
    grouped_data[key].append(value)

# 打印分组结果
for key in grouped_data:
    print(key, ':', grouped_data[key])

这个示例代码定义了一个包含6个元素的列表data，列表中每个元素都是一个由键和值组成的元组，表示一条数据。这个示例中需要按照键值进行分组，分组结果使用defaultdict函数进行存储，然后通过遍历分组结果输出分组后的数据。执行这个示例代码可以得到以下结果：

color : ['red', 'blue']
fruit : ['apple', 'orange']
animal : ['elephant', 'lion']

下面是一个实现键值分区的复杂示例代码：

import pandas as pd

# 构造数据
data = pd.DataFrame({
    'name': ['Alice', 'Bob', 'Charlie', 'David', 'Ella', 'Frank', 'Grace', 'Henry'],
    'age': [25, 36, 28, 19, 27, 32, 25, 21],
    'gender': ['F', 'M', 'M', 'F', 'F', 'M', 'F', 'M'],
    'score': [80, 90, 85, 75, 95, 90, 85, 80]
})

# 使用groupby函数按照年龄分成两组，小于30岁一组，大于等于30岁为一组
grouped_data = data.groupby(data['age'] < 30)

# 打印分区结果
for key in grouped_data.groups:
    print(key)
    print(grouped_data.get_group(key))

这个示例代码使用了pandas库中的groupby函数，将数据按照年龄分成小于30岁和大于等于30岁两组，然后通过遍历分区结果输出分区后的数据。执行这个示例代码可以得到以下结果：

False
   name  age gender  score
1   Bob   36      M     90
5  Frank   32      M     90
True
      name  age gender  score
0    Alice   25      F     80
2  Charlie   28      M     85
3    David   19      F     75
4     Ella   27      F     95
6    Grace   25      F     85
7    Henry   21      M     80

通过以上示例代码，可以了解Python中使用defaultdict函数和pandas库中的groupby函数实现键值分组或分区数据的不同方法，这些方法在实际开发中都有广泛的应用场景。

Tags: