使用itertools groupby从分类列表中获取索引值

import itertools

def get_index_by_category(category_list):
    # 对分类列表进行排序，确保相同分类的元素相邻
    sorted_list = sorted(category_list)

    # 使用itertools.groupby进行分组
    grouped_data = itertools.groupby(sorted_list)

    # 遍历分组数据，获取索引值
    index_by_category = {}
    index = 0
    for key, group in grouped_data:
        # 将每个分类的索引值存储到字典中
        index_by_category[key] = index
        # 更新索引值
        index += len(list(group))

    return index_by_category

# 示例用法
category_list = ['A', 'A', 'B', 'B', 'B', 'C', 'D', 'D']
index_by_category = get_index_by_category(category_list)
print(index_by_category)

{'A': 0, 'B': 2, 'C': 5, 'D': 6}