Python中是否有一个内置函数可以像C++一样获得列表的大小

Question

当一个列表只有一个维度时，len()可以用来获取它的大小，有没有函数可以获取嵌套列表的大小？

例如，test_list = [[1], [2], [3]]，len(test_list)返回3，如何获得(3, 1)？

Answer 1

如果使用numpy数组而不是list，则可以获得(3，1(的形状。

import numpy as np
a = np.array([[1], [2], [3]])
print(a.shape)

Answer 2

这是因为数组的元素数量实际上是3，如果你想检查其内容的深度，你可以这样做(注意，第二维度的深度将是其内容的最大长度(

test_list = [[1], [2], [3]]
print((len(test_list), max(len(i) for i in test_list))) # -> (3, 1)

或使用numpy

import numpy as np
test_list = np.array([[1], [2], [3]])
print(test_list.shape) # -> (3, 1)