Python中实现获取形状列表的关键函数get_shape_list()解析

在Python中，可以使用get_shape_list()函数来获取形状列表。该函数的主要功能是从给定的输入列表中筛选出所有形状，并将其添加到形状列表中。下面是该函数的实现和使用示例：

def get_shape_list(input_list):
    shape_list = []  # 创建一个空的形状列表

    for item in input_list:
        if isinstance(item, int) or isinstance(item, float):
            shape_list.append('Scalar')  # 如果当前元素是数字，则将其添加到形状列表中的标量部分
        elif isinstance(item, str):
            shape_list.append('String')  # 如果当前元素是字符串，则将其添加到形状列表中的字符串部分
        elif isinstance(item, list):
            if isinstance(item[0], list):
                shape_list.append('Matrix')  # 如果当前元素是嵌套列表，则将其添加到形状列表中的矩阵部分
            else:
                shape_list.append('Vector')  # 如果当前元素是一维列表，则将其添加到形状列表中的向量部分

    return shape_list

上述函数首先创建一个空的形状列表shape_list。然后，对于给定的输入列表中的每个元素，使用if-elif-else条件分支检查元素的类型。如果元素是整数或浮点数，则将其添加到形状列表的标量部分；如果元素是字符串，则将其添加到字符串部分；如果元素是嵌套的二维列表，则将其添加到矩阵部分；如果元素是一维列表，则将其添加到向量部分。最后，函数返回形状列表。

以下是get_shape_list()函数的使用示例：

input_list = [1, 2, ['a', 'b'], 'string', [[1, 2], [3, 4]], [5, 6, 7]]
shape_list = get_shape_list(input_list)
print(shape_list)

输出结果为：

['Scalar', 'Scalar', 'Vector', 'String', 'Matrix', 'Vector']

在上述示例中，输入列表input_list包含了不同类型的元素，包括整数、字符串和列表。通过调用get_shape_list()函数，我们获取了该列表的形状列表，并将其打印输出。输出结果显示了每个元素的形状，其中的标量、字符串、向量和矩阵部分分别用对应的字符串表示。