检查元素的json数组并返回与该元素关联的完整数组

import json

def find_related_arrays(json_array, target_element):
    related_arrays = []

    def search_array(array, target):
        for item in array:
            if item == target:
                return True
            elif isinstance(item, list):
                if search_array(item, target):
                    return True
            elif isinstance(item, dict):
                if search_array(item.values(), target):
                    return True
        return False

    for element in json_array:
        if isinstance(element, list):
            if search_array(element, target_element):
                related_arrays.append(element)
        elif isinstance(element, dict):
            if search_array(element.values(), target_element):
                related_arrays.append(element)

    return related_arrays

# 示例用法
json_str = '[{"id": 1, "name": "John", "friends": [2, 3]}, {"id": 2, "name": "Alice", "friends": [1, 3]}, {"id": 3, "name": "Bob", "friends": [1, 2]}]'
json_array = json.loads(json_str)
target_element = 2

related_arrays = find_related_arrays(json_array, target_element)
print(related_arrays)