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)