用户画像关联分析

from itertools import combinations
from collections import defaultdict

# 模拟用户购买商品的数据
data = [
    ['user1', 'itemA', 'itemB'],
    ['user2', 'itemA', 'itemC'],
    ['user3', 'itemB', 'itemD'],
    ['user4', 'itemA', 'itemB', 'itemD']
]

# 统计商品组合出现的次数
item_combinations = defaultdict(int)
for user_items in data:
    for combo in combinations(user_items[1:], 2):
        item_combinations[combo] += 1

# 输出常见的商品组合
for combo, count in sorted(item_combinations.items(), key=lambda x: x[1], reverse=True):
    print(f"{combo}: {count} 次")