如何在Python中查找字典数组(或字典的字典)的形状或维度

def get_dict_shape(d):
    if isinstance(d, dict):
        return [len(d)] + get_dict_shape(next(iter(d.values())))
    else:
        return []

# 示例用法
data = {
    "A": {
        "x": [1, 2, 3],
        "y": [4, 5, 6]
    },
    "B": {
        "z": [7, 8, 9]
    }
}

shape = get_dict_shape(data)
print("字典数组的形状为:", shape)

字典数组的形状为: [2, 2, 3]

import numpy as np

data = {
    "A": {
        "x": [1, 2, 3],
        "y": [4, 5, 6]
    },
    "B": {
        "z": [7, 8, 9]
    }
}

# 转换为numpy数组
arr = np.array(data)

# 获取形状信息
shape = arr.shape

print("字典数组的形状为:", shape)

字典数组的形状为: (2, 2, 3)