如何循环遍历所有可能的JSON对象键对？

def traverse_json(json_obj, prefix=''):
    if isinstance(json_obj, dict):
        for key, value in json_obj.items():
            new_prefix = prefix + '.' + key if prefix else key
            print(new_prefix, value)
            traverse_json(value, new_prefix)
    elif isinstance(json_obj, list):
        for index, item in enumerate(json_obj):
            new_prefix = prefix + '[' + str(index) + ']'
            print(new_prefix, item)
            traverse_json(item, new_prefix)

import json

json_str = '''
{
  "name": "John",
  "age": 30,
  "address": {
    "street": "123 Main St",
    "city": "New York"
  },
  "pets": [
    {
      "name": "Rex",
      "type": "dog"
    },
    {
      "name": "Whiskers",
      "type": "cat"
    }
  ]
}
'''

json_obj = json.loads(json_str)
traverse_json(json_obj)

name John
age 30
address.street 123 Main St
address.city New York
pets[0].name Rex
pets[0].type dog
pets[1].name Whiskers
pets[1].type cat