基于计数对记录集进行分类

pip install pandas

import pandas as pd

# 示例数据
data = {
    'category': ['A', 'B', 'A', 'C', 'B', 'A', 'C', 'C', 'B', 'A'],
    'value': [10, 20, 10, 30, 20, 10, 30, 30, 20, 10]
}

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

# 计算每个类别的计数
category_counts = df['category'].value_counts().reset_index()
category_counts.columns = ['category', 'count']

# 定义分类规则
def classify(count):
    if count >= 4:
        return 'High'
    elif count >= 2:
        return 'Medium'
    else:
        return 'Low'

# 应用分类规则
category_counts['classification'] = category_counts['count'].apply(classify)

# 合并分类结果回原始DataFrame
df = df.merge(category_counts[['category', 'classification']], on='category', how='left')

# 打印结果
print(df)

  category  value classification
0        A     10         Medium
1        B     20         Medium
2        A     10         Medium
3        C     30           High
4        B     20         Medium
5        A     10         Medium
6        C     30           High
7        C     30           High
8        B     20         Medium
9        A     10         Medium