如何将不规则嵌套列表的元素映射为指定字典的对应value值

核心思路 #

你的嵌套列表为不规则多层结构，此前的方法仅支持单层映射，无法递归处理深层嵌套的子元素，因此需要通过递归遍历的方式实现逐层替换。

具体实现 #

步骤1：定义递归映射函数和映射字典 #

# 定义类别映射字典
label_map = {'checkbox': 0,
 'radio': 1,
 'dropdown': 2,
 'listbox': 3,
 'button': 4,
 'link': 5,
 'input': 6,
 'label': 7}

# 递归处理函数
def remap_nested(item, mapping):
    # 如果当前元素是列表，递归处理每个子元素
    if isinstance(item, list):
        return [remap_nested(sub_item, mapping) for sub_item in item]
    # 如果是标签字符串，直接返回对应映射值
    else:
        return mapping[item]

步骤2：不同场景调用方式 #

• 场景1：直接处理Python原生嵌套列表

# 示例不规则嵌套列表
raw_labels = [
    'link',
    'dropdown',
    'dropdown',
    ['input', 'link'],
    ['input', 'button', 'link'],
    ['dropdown', 'button']
]

# 执行映射
remapped_labels = remap_nested(raw_labels, label_map)

运行后输出结果：

[5, 2, 2, [6, 5], [6, 4, 5], [2, 4]]

• 场景2：处理Pandas DataFrame中的标签列

import pandas as pd

# 假设df是你的原始数据框，labels列为嵌套列表
df['labels_remapped'] = df['labels'].apply(lambda x: remap_nested(x, label_map))

方法失效原因说明 #

你此前尝试的方案无法生效的核心问题：

1. df['labels'].replace(Dict) 是将整个子列表作为匹配对象，不会遍历子列表内部元素做替换
2. 直接用map(Dict, 子列表)仅支持单层子列表映射，遇到深层嵌套的子元素会抛出类型错误

内容的提问来源于stack exchange，提问作者Gustavo Scholze