欢迎访问宙启技术站
智能推送

Python中get_shape_list()函数的原理与实现

发布时间:2024-01-09 16:16:41

get_shape_list()函数的原理是获取一个列表中所有嵌套列表的形状(shape),即获取嵌套列表中每个子列表的长度。

函数的实现步骤如下:

1. 定义一个空列表shape_list,用于存储每个子列表的长度。

2. 遍历输入的列表。

3. 对于每个子列表,获取其长度,并将长度添加到shape_list中。

4. 返回shape_list作为输出结果。

下面是一个使用get_shape_list()函数的例子:

def get_shape_list(lst):
    shape_list = []
    for sublist in lst:
        shape_list.append(len(sublist))
    return shape_list

# 示例输入
my_list = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]

# 调用get_shape_list函数获取嵌套列表形状
result = get_shape_list(my_list)

# 打印结果
print(result) 

在上面的例子中,输入的my_list是一个嵌套列表,包含3个子列表。通过调用get_shape_list()函数,获取了每个子列表的长度,并将结果存储在变量result中。

输出结果为[3, 2, 4],即表示my_list中的 个子列表长度为3,第二个子列表长度为2,第三个子列表长度为4。