Python3中简化遍历JSON列表从指定键获取后续键的方法

首先，咱们先拆解下你现有代码里的核心问题：

• 你错误地用键的长度去作为列表的索引（比如len(list(token.keys())[0])），这完全不符合逻辑——列表的索引是元素的位置（0、1、2...），和键的长度没关系，这是导致输出混乱的根本原因。
• 嵌套循环和错误的切片操作让代码逻辑变得复杂冗余，还容易引发不必要的错误。

接下来，咱们用更简洁、低复杂度的方式实现需求，整体时间复杂度是O(n)（仅需线性遍历列表两次）：

实现思路 #

1. 先遍历列表，定位到包含目标键的元素的位置索引。
2. 从该索引的下一个位置开始，遍历后续所有元素，提取每个元素的唯一键。
3. 把收集到的键按你期望的格式输出（逗号分隔）。

优化后的代码 #

# 你的原始数据列表
dictionary = [{"a": "1"}, {"b": "2"}, {"c": "3"}, {"d": "4"}]
target_key = "b"  # 给定的起始键

try:
    # 第一步：找到目标键所在元素的索引
    start_index = None
    for idx, item in enumerate(dictionary):
        if target_key in item:
            start_index = idx
            break
    
    if start_index is None:
        print(f"键'{target_key}'不存在于列表中")
    else:
        # 第二步：从起始索引的下一个位置开始，收集所有键
        result_keys = []
        # 直接切片获取目标位置之后的所有元素
        for item in dictionary[start_index+1:]:
            # 高效提取单键字典的键（无需转换为列表）
            key = next(iter(item.keys()))
            result_keys.append(key)
        
        # 第三步：按期望格式输出
        print(", ".join(result_keys))
except Exception as error:
    print(f"发生错误：{str(error)}")

关键细节说明 #

• enumerate(dictionary)：同时获取元素的索引和值，精准定位目标键的位置。
• next(iter(item.keys()))：比list(item.keys())[0]更高效，因为不需要把键转换成列表，直接迭代取第一个键。
• dictionary[start_index+1:]：通过切片直接获取目标位置之后的所有元素，避免多余的循环判断。

测试结果 #

当target_key = "b"时，输出为：c, d，完全符合你的期望。如果目标键不存在（比如target_key = "x"），程序会输出友好的提示信息，避免崩溃。

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