Python查找列表中每个子列表的大小

# 定义一个二维列表
matrix = [
    [1, 2, 3],
    [4, 5],
    [6, 7, 8, 9]
]

# 使用列表推导式获取每个子列表的大小
sizes = [len(sublist) for sublist in matrix]

print(sizes)  # 输出: [3, 2, 4]

# 检查并打印非空子列表的大小
non_empty_sizes = [len(sublist) for sublist in matrix if len(sublist) > 0]
print(non_empty_sizes)  # 输出: [3, 2, 4]

def get_sublist_sizes(lst):
    sizes = []
    for item in lst:
        if isinstance(item, list):
            sizes.append(get_sublist_sizes(item))
        else:
            sizes.append(1)
    return sizes

# 定义一个三维列表
cube = [
    [[1, 2], [3, 4]],
    [[5, 6, 7], [8]]
]

print(get_sublist_sizes(cube))  # 输出: [[2, 2], [3, 1]]