嵌套对象键上的regex以过滤列表python

import re

def filter_keys(obj, pattern):
    if isinstance(obj, dict):
        return {k: filter_keys(v, pattern) for k, v in obj.items() if re.search(pattern, k)}
    elif isinstance(obj, list):
        return [filter_keys(item, pattern) for item in obj]
    else:
        return obj

# 示例数据
data = {
    'key1': 'value1',
    'nested': {
        'key2': 'value2',
        'key3': 'value3'
    },
    'list': [
        {'key4': 'value4'},
        {'key5': 'value5'}
    ]
}

# 使用正则表达式过滤键
filtered_data = filter_keys(data, r'key[2-4]')

print(filtered_data)

{
    'nested': {
        'key2': 'value2',
        'key3': 'value3'
    },
    'list': [
        {'key4': 'value4'}
    ]
}