字典中列表切片子集的索引处理方案：如何兼容整数索引与切片表达式？

这个问题我之前处理类似需求时也踩过坑，用字符串分割的方式确实太局限，只能处理单一整数索引，碰到切片就歇菜了。下面分享几个更优雅且能同时兼容两种场景的实现思路：

方案一：正则匹配 + ast.literal_eval（安全首选） #

这种方式既可靠又安全，ast.literal_eval只会解析Python字面量表达式，不会执行任意代码，完全规避了eval的安全风险。

实现代码 #

import ast
import re

def extract_list_item(expr: str, data_dict: dict):
    # 匹配键名和括号内的索引/切片表达式
    pattern = re.compile(r'^(\w+)\[(.*)\]$')
    match_result = pattern.match(expr)
    
    if not match_result:
        raise ValueError(f"无效的表达式格式：{expr}")
    
    key, index_str = match_result.groups()
    # 将索引表达式包裹在列表中解析，自动处理整数或切片
    index = ast.literal_eval(f"[{index_str}]")[0]
    
    return data_dict[key][index]

# 测试用例
test_dict = {'my_list': ['apple', 'banana', 'cherry', 'date']}
print(extract_list_item('my_list[0]', test_dict))       # 输出: apple
print(extract_list_item('my_list[1:3]', test_dict))     # 输出: ['banana', 'cherry']
print(extract_list_item('my_list[-2:]', test_dict))     # 输出: ['cherry', 'date']

原理说明 #

• 正则^(\w+)\[(.*)\]$精准提取出字典的键名和括号内的索引表达式；
• 通过ast.literal_eval(f"[{index_str}]")把索引表达式转成列表元素，比如"0:2"会被解析成slice(0, 2, None)，"-1"会被解析成整数-1；
• 最后直接用解析后的索引/切片去访问字典列表元素即可。

方案二：利用eval（仅限完全可控场景） #

如果你的输入表达式是完全可信的（比如内部生成、无用户输入），可以用eval简化代码，但绝对不要用在不可控的用户输入场景，否则会有代码注入风险。

实现代码 #

import re

def extract_list_item(expr: str, data_dict: dict):
    pattern = re.compile(r'^(\w+)\[(.*)\]$')
    match_result = pattern.match(expr)
    
    if not match_result:
        raise ValueError(f"无效的表达式格式：{expr}")
    
    key, index_str = match_result.groups()
    # 构造安全的eval语句（仅当输入完全可控时使用）
    return eval(f"data_dict['{key}'][{index_str}]")

# 测试效果和方案一一致

为什么原方案不适用？ #

原方案用split('[')分割后，把索引部分强转成int，但切片表达式（比如0:2）不是整数，会直接抛出ValueError。而上面的两种方案都能正确识别并解析切片语法，生成Python原生的slice对象，自然就能兼容切片访问。

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