使用Python的get_shape_list()函数获取形状列表

get_shape_list()函数用于获取一个列表中所有元素的形状。它的输入参数是一个列表，返回值是一个包含各个元素的形状的新列表。以下是一个使用Python编写的示例代码，展示了如何使用该函数。

import numpy as np

def get_shape_list(lst):
    shape_list = []
    for i in lst:
        shape_list.append(np.shape(i))
    return shape_list

# 示例列表
example_list = [np.array([[1, 2, 3], [4, 5, 6]]),
                np.array([1, 2, 3]), 
                np.array([[1, 2], [3, 4], [5, 6]]),
                np.array([[[1], [2]], [[3], [4]], [[5], [6]]]),
                np.array([1])]

# 使用get_shape_list()函数获取形状列表
shapes = get_shape_list(example_list)

# 打印结果
for i, shape in enumerate(shapes):
    print(f"Shape of element {i+1}: {shape}")

输出结果:

Shape of element 1: (2, 3)
Shape of element 2: (3,)
Shape of element 3: (3, 2)
Shape of element 4: (3, 2, 1)
Shape of element 5: (1,)

在上述示例中，我们首先导入了numpy库，并定义了一个名为get_shape_list()的函数。该函数使用了numpy的shape函数来获取每个元素的形状，并将结果添加到一个新的形状列表中。最后，我们使用示例列表调用get_shape_list()函数，将结果打印出来。

上述示例中的示例列表包含了不同形状的numpy数组。其中， 个元素是一个2行3列的二维数组，第二个元素是一个长度为3的一维数组，第三个元素是一个3行2列的二维数组，第四个元素是一个3行2列的三维数组，而第五个元素是一个包含单个元素的一维数组。通过调用get_shape_list()函数，我们获取了每个元素的形状，并将结果打印出来。

如果你想使用get_shape_list()函数获取其他列表中元素的形状，只需要将你的列表传递给get_shape_list()函数即可。这个函数可以适用于任何列表，并返回一个包含各个元素形状的新列表。