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'}
     ]},
]

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

区域: EU, 用户ID: 1
区域: EU, 用户ID: 2
区域: NA, 用户ID: 1
区域: NA, 用户ID: 2
区域: NA, 用户ID: 3

class NestIterator:
    """
    一个用于遍历嵌套数据结构的自定义迭代器。
    它封装了遍历逻辑，允许用户通过指定键来提取所需的数据。
    """
    def __init__(self, data_source, *keys):
        """
        初始化迭代器。
        :param data_source: 要遍历的嵌套数据（例如，上述的 `data` 列表）。
        :param keys: 一个元组，包含用于访问嵌套数据的键。
                     例如，('region', 'users', 'id') 表示从 'region' 获取区域，
                     然后进入 'users' 列表，最后从用户字典中获取 'id'。
        """
        self._data = data_source
        self._keys = keys

    def __iter__(self):
        """
        使类成为一个迭代器，返回自身。
        """
        return self._traverse()

    def _traverse(self):
        """
        实际的遍历逻辑。
        这是一个生成器方法，按需产出数据。
        """
        # 假设 _keys 至少包含三个元素：
        # _keys[0] 用于第一个字典的键（如 'region'）
        # _keys[1] 用于第二个列表的键（如 'users'）
        # _keys[2] 用于第二个字典的键（如 'id' 或 'name'）

        # 针对当前数据结构定制的遍历逻辑
        for item1 in self._data:
            # 检查 item1[self._keys[1]] 是否存在且可迭代
            if self._keys[1] in item1 and isinstance(item1[self._keys[1]], list):
                for item2 in item1[self._keys[1]]:
                    # 确保 item2 是字典且包含目标键
                    if isinstance(item2, dict) and self._keys[2] in item2:
                        yield (item1[self._keys[0]], item2[self._keys[2]])
                    else:
                        # 可选：处理item2不是字典或缺少键的情况
                        pass
            else:
                # 可选：处理item1缺少第二个键或其值不是列表的情况
                pass

# 使用自定义迭代器
print("\n--- 使用自定义迭代器提取 ID ---")
for item in NestIterator(data, 'region', 'users', 'id'):
    print(f'区域: {item[0]}, 用户ID: {item[1]}')

print("\n--- 使用自定义迭代器提取 Name ---")
for item in NestIterator(data, 'region', 'users', 'name'):
    print(f'区域: {item[0]}, 用户名: {item[1]}')

--- 使用自定义迭代器提取 ID ---
区域: EU, 用户ID: 1
区域: EU, 用户ID: 2
区域: NA, 用户ID: 1
区域: NA, 用户ID: 2
区域: NA, 用户ID: 3

--- 使用自定义迭代器提取 Name ---
区域: EU, 用户名: xyz
区域: EU, 用户名: foo
区域: NA, 用户名: bar
区域: NA, 用户名: foo
区域: NA, 用户名: foo