python - 如何确定列表的维度？

Question

如何以编程方式确定列表的维度？谢谢。

Answer 1

对于一个简单的列表，您可以通过以下方式获取其长度len：

>>> l = [1, 2, 3, 4, 5]
>>> len(l)
5

对于最容易表示为嵌套列表的矩阵，您可以获得第一个子列表的长度，例如：

>>> matrix = [
    [1, 2],
    [3, 4],
    [5, 6],
    [7, 8],
    [9, 10]]
>>> len(matrix)
5
>>> len(matrix[0])
2

考虑到 JBernardo 的评论，定义一个简单的iterable助手：

>>> def iterable(x):
    if isinstance(x, basestring): return False
    try:
        iter(x)
    except TypeError:
        return False
    return True

>>> iterable(4)
False
>>> iterable([1, 2, 3])
True

然后我们可以递归地定义dimensionality函数：

>>> def dimensionality(l):
    if not iterable(l): return 0
    return 1 + dimensionality(l[0])

>>> dimensionality(0)
0
>>> dimensionality([1, 2, 3])
1
>>> dimensionality([[1,2], [2,3], [3,4]])
2
>>> dimensionality([[[1,2],[2,3]], [[2,3],[3,4]], [[3,4],[4,5]]])
3

而不是iterable你可以做isinstance(x, list)或你想要的任何其他检查。编辑排除字符串以避免dimensionality('lol')无限循环。

Answer 2

假设均匀性：

dims = []
while isinstance(matrix, list) and matrix is not None:
    dims.append(len(matrix))
    matrix = matrix[0]
number_of_dimensions = len(dims)

希望有帮助