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)

希望有所帮助

如何确定列表的维度？

2 个答案: