Python for array中的"GROUP BY“函数

from itertools import groupby

# 假设有一个包含字典的数组
data = [
    {'name': 'Alice', 'age': 25, 'gender': 'female'},
    {'name': 'Bob', 'age': 30, 'gender': 'male'},
    {'name': 'Charlie', 'age': 25, 'gender': 'male'},
    {'name': 'Dave', 'age': 30, 'gender': 'male'},
    {'name': 'Eve', 'age': 25, 'gender': 'female'}
]

# 根据年龄进行分组
data.sort(key=lambda x: x['age'])  # 先按照年龄排序
groups = groupby(data, key=lambda x: x['age'])  # 根据年龄分组

# 遍历分组结果
for age, group in groups:
    print(f"Age: {age}")
    for item in group:
        print(f"Name: {item['name']}, Gender: {item['gender']}")
    print()

Age: 25
Name: Alice, Gender: female
Name: Charlie, Gender: male
Name: Eve, Gender: female

Age: 30
Name: Bob, Gender: male
Name: Dave, Gender: male

import pandas as pd

# 假设有一个包含字典的数组
data = [
    {'name': 'Alice', 'age': 25, 'gender': 'female'},
    {'name': 'Bob', 'age': 30, 'gender': 'male'},
    {'name': 'Charlie', 'age': 25, 'gender': 'male'},
    {'name': 'Dave', 'age': 30, 'gender': 'male'},
    {'name': 'Eve', 'age': 25, 'gender': 'female'}
]

# 创建DataFrame对象
df = pd.DataFrame(data)

# 根据年龄进行分组
groups = df.groupby('age')

# 遍历分组结果
for age, group in groups:
    print(f"Age: {age}")
    for index, row in group.iterrows():
        print(f"Name: {row['name']}, Gender: {row['gender']}")
    print()