如何在json对象上创建具有每个排列组合json对象数组

import copy

def generate_combinations(json_obj):
    combinations = []
    keys = list(json_obj.keys())
    values = list(json_obj.values())
    num_keys = len(keys)
    num_values = len(values[0])
    
    def generate_combinations_helper(curr_combination, index):
        if index == num_keys:
            combinations.append(copy.deepcopy(curr_combination))
            return
        
        for i in range(num_values):
            curr_combination[keys[index]] = values[index][i]
            generate_combinations_helper(curr_combination, index + 1)
    
    generate_combinations_helper({}, 0)
    return combinations

# 示例输入
json_obj = {
    "color": ["red", "blue"],
    "size": ["small", "large"],
    "shape": ["circle", "square"]
}

combinations = generate_combinations(json_obj)
print(combinations)

[
    {"color": "red", "size": "small", "shape": "circle"},
    {"color": "red", "size": "small", "shape": "square"},
    {"color": "red", "size": "large", "shape": "circle"},
    {"color": "red", "size": "large", "shape": "square"},
    {"color": "blue", "size": "small", "shape": "circle"},
    {"color": "blue", "size": "small", "shape": "square"},
    {"color": "blue", "size": "large", "shape": "circle"},
    {"color": "blue", "size": "large", "shape": "square"}
]