如何在python中打印字典函数中的数组形状

pip install numpy

import numpy as np

def print_array_shape(dictionary):
    for key, value in dictionary.items():
        if isinstance(value, np.ndarray):
            print(f"The shape of array {key} is: {value.shape}")
        else:
            print(f"{key} is not an array.")

# 示例字典
my_dict = {
    'array1': np.array([1, 2, 3]),
    'array2': np.array([[1, 2, 3], [4, 5, 6]])
}

print_array_shape(my_dict)