Python嵌套数据结构的高效与优雅遍历：自定义迭代器模式

data = [
    {'region': 'EU',
     'users' : [
         { 'id': 1, 'name': 'xyz'},
         { 'id': 2, 'name': 'foo'}
    ]},
    {'region': 'NA',
     'users' : [
         { 'id': 1, 'name': 'bar'},
         { 'id': 2, 'name': 'foo'},
         { 'id': 3, 'name': 'foo'}
    ]},
]

for region_data in data:
    for user_data in region_data['users']:
        print(f'区域: {region_data["region"]}, 用户ID: {user_data["id"]}')

class NestIterator:
    """
    一个用于遍历特定两层嵌套数据结构的自定义迭代器。
    它接受原始数据和一个包含三个键的元组：
    1. 外部字典中需要提取的键（例如 'region'）。
    2. 外部字典中包含内部列表的键（例如 'users'）。
    3. 内部字典中需要提取的键（例如 'id' 或 'name'）。
    """
    def __init__(self, data, *keys):
        self._data = data
        self._keys = keys

    def __iter__(self):
        """
        使类实例可迭代，返回一个生成器。
        """
        return self._traverse()

    def _traverse(self):
        """
        实际的遍历逻辑，使用yield生成器逐个返回结果。
        """
        # 确保传入的键数量符合预期
        if len(self._keys) < 3:
            raise ValueError("NestIterator requires at least three keys: outer_value_key, inner_list_key, inner_value_key")

        outer_value_key = self._keys[0]
        inner_list_key = self._keys[1]
        inner_value_key = self._keys[2]

        for outer_item in self._data:
            # 检查外部项是否为字典，以及是否包含内部列表的键
            if isinstance(outer_item, dict) and inner_list_key in outer_item and isinstance(outer_item[inner_list_key], list):
                for inner_item in outer_item[inner_list_key]:
                    # 检查内部项是否为字典，以及是否包含所需的值键
                    if isinstance(inner_item, dict) and outer_value_key in outer_item and inner_value_key in inner_item:
                        yield (outer_item[outer_value_key], inner_item[inner_value_key])

# 原始数据
data = [
    {'region': 'EU',
     'users' : [
         { 'id': 1, 'name': 'xyz'},
         { 'id': 2, 'name': 'foo'}
    ]},
    {'region': 'NA',
     'users' : [
         { 'id': 1, 'name': 'bar'},
         { 'id': 2, 'name': 'foo'},
         { 'id': 3, 'name': 'foo'}
    ]},
]

print("--- 遍历区域和用户ID ---")
for region, user_id in NestIterator(data, 'region', 'users', 'id'):
    print(f'区域: {region}, 用户ID: {user_id}')

print("\n--- 遍历区域和用户名 ---")
for region, user_name in NestIterator(data, 'region', 'users', 'name'):
    print(f'区域: {region}, 用户名: {user_name}')

--- 遍历区域和用户ID ---
区域: EU, 用户ID: 1
区域: EU, 用户ID: 2
区域: NA, 用户ID: 1
区域: NA, 用户ID: 2
区域: NA, 用户ID: 3

--- 遍历区域和用户名 ---
区域: EU, 用户名: xyz
区域: EU, 用户名: foo
区域: NA, 用户名: bar
区域: NA, 用户名: foo
区域: NA, 用户名: foo