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

在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]

利用numpy库进行形状计算：
- 首先，将字典数组转换为numpy数组对象。
- 使用numpy库的shape属性获取数组的形状信息。
- 以下是一个示例代码：

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)

输出结果为：