在python中查找嵌套列表的维度

时间:2014-10-17 14:34:21

标签: python algorithm matrix multidimensional-array

我们可以使用嵌套列表在python中创建多维数组,例如:

A = [[1,2,3],
     [2,1,3]]

等。 在这种情况下,简单的是nRows = len(A)和nCols = len(A [0])。但是,当我有三个以上的维度时,它会变得复杂。

A = [[[1,1,[1,2,3,4]],2,[3,[2,[3,4]]]],
     [2,1,3]]

等。 这些列表在Python中是合法的。并且维度的数量不是先验的。 在这种情况下,如何确定每个维度中的维度数和元素数。

我正在寻找一种算法,如果可能的话。我相信它有类似于DFS的东西。有什么建议吗?

P.S。:我不是在寻找任何现有的套餐,但我想了解它们。

1 个答案:

答案 0 :(得分:0)

我相信能解决我自己的问题。 这只是一个简单的DFS。

对于上面给出的例子:A = [[[1,1,[1,2,3,4]],2,[3,[2,[3,4]]]],      [2,1,3]] 答案如下: [[3,2,2,3,3,4],[3]] 维度总数为7.

我想我是在过度思考......不管怎样,谢谢......!